Podtekatel commited on
Commit
4cee7a5
1 Parent(s): 08befe8

Update README

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -48,10 +48,11 @@ article = "This is one of my successful experiments on style transfer. I've buil
48
  "" \
49
  "Model pipeline which used in project is improved CartoonGAN.<br>" \
50
  "This model was trained on RTX 2080 Ti 1.5 days with batch size 7.<br>" \
 
51
  "If you want to use this app or integrate this model into yours, please contact me at email '[email protected]'."
52
 
53
  imgs_folder = 'demo'
54
- examples = [[os.path.join(imgs_folder, img_filename)] for img_filename in os.listdir(imgs_folder)]
55
 
56
  demo = gr.Interface(
57
  fn=inference,
 
48
  "" \
49
  "Model pipeline which used in project is improved CartoonGAN.<br>" \
50
  "This model was trained on RTX 2080 Ti 1.5 days with batch size 7.<br>" \
51
+ "Model weights 64 MB in ONNX fp32 format, infers 25 ms on GPU and 150 ms on CPU at 256x256 resolution.<br>" \
52
  "If you want to use this app or integrate this model into yours, please contact me at email '[email protected]'."
53
 
54
  imgs_folder = 'demo'
55
+ examples = [[os.path.join(imgs_folder, img_filename)] for img_filename in sorted(os.listdir(imgs_folder))]
56
 
57
  demo = gr.Interface(
58
  fn=inference,