Dollaya Piumsuwan commited on
Commit
3ab007c
·
verified ·
1 Parent(s): 951f0ee

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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}