Spaces:
Build error
Build error
Podtekatel
commited on
Commit
•
4cee7a5
1
Parent(s):
08befe8
Update README
Browse files
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,
|