Spaces:
Runtime error
Runtime error
share true
Browse files- app/app.py +2 -3
app/app.py
CHANGED
|
@@ -11,9 +11,8 @@ import matplotlib.pyplot as plt
|
|
| 11 |
import numpy as np
|
| 12 |
import torch
|
| 13 |
from albumentations.pytorch import ToTensorV2
|
| 14 |
-
from PIL import Image
|
| 15 |
-
|
| 16 |
from model import Classifier
|
|
|
|
| 17 |
|
| 18 |
# Load the model
|
| 19 |
model = Classifier.load_from_checkpoint("./models/checkpoint.ckpt")
|
|
@@ -88,7 +87,7 @@ def app():
|
|
| 88 |
"./test_images/elephant.jpg",
|
| 89 |
"./test_images/horse.jpeg",
|
| 90 |
],
|
| 91 |
-
).launch()
|
| 92 |
|
| 93 |
|
| 94 |
# Run the app
|
|
|
|
| 11 |
import numpy as np
|
| 12 |
import torch
|
| 13 |
from albumentations.pytorch import ToTensorV2
|
|
|
|
|
|
|
| 14 |
from model import Classifier
|
| 15 |
+
from PIL import Image
|
| 16 |
|
| 17 |
# Load the model
|
| 18 |
model = Classifier.load_from_checkpoint("./models/checkpoint.ckpt")
|
|
|
|
| 87 |
"./test_images/elephant.jpg",
|
| 88 |
"./test_images/horse.jpeg",
|
| 89 |
],
|
| 90 |
+
).launch(share=True)
|
| 91 |
|
| 92 |
|
| 93 |
# Run the app
|