Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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(
|
|
|
|
| 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()
|