Spaces:
Sleeping
Sleeping
Dollaya Piumsuwan commited on
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -16,7 +16,7 @@ import io
|
|
| 16 |
# SETTINGS
|
| 17 |
# ########################
|
| 18 |
model_file = "models/momaclassifier_resnet50.pt"
|
| 19 |
-
image_csv = "demo_artworks.csv"
|
| 20 |
image_folder = "demo_images"
|
| 21 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 22 |
class_index = {"Drawing": 0, "Photograph": 1, "Print": 2}
|
|
|
|
| 16 |
# SETTINGS
|
| 17 |
# ########################
|
| 18 |
model_file = "models/momaclassifier_resnet50.pt"
|
| 19 |
+
image_csv = "data/demo_artworks.csv"
|
| 20 |
image_folder = "demo_images"
|
| 21 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 22 |
class_index = {"Drawing": 0, "Photograph": 1, "Print": 2}
|