hieupt commited on
Commit
4effec7
·
1 Parent(s): a5b1479

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,6 +7,7 @@ from typing import Tuple, Dict
7
  from model import TransformerNet
8
  from torchvision import transforms
9
  from PIL import Image
 
10
 
11
  # Get model
12
  model_dir = '/models'
@@ -71,4 +72,4 @@ demo = gr.Interface(fn=transfer, # mapping function from input to output
71
  article=article)
72
 
73
  # Launch the demo!
74
- demo.launch(debug=True)
 
7
  from model import TransformerNet
8
  from torchvision import transforms
9
  from PIL import Image
10
+ from pathlib import Path
11
 
12
  # Get model
13
  model_dir = '/models'
 
72
  article=article)
73
 
74
  # Launch the demo!
75
+ demo.launch()