Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,13 +11,18 @@ import torch.nn as nn
 from PIL import Image
 import os
 import io
+from pathlib import Path
 
 #########################
 # SETTINGS
 # ########################
-model_file = "models/momaclassifier_resnet50.pt"
-image_csv = "data/demo_artworks.csv"
-image_folder = "demo_images"
+base_path = Path(__file__).resolve().parent.parent
+model_file = base_path / "models" / "momaclassifier_resnet50.pt"
+image_csv = base_path / "data" / "demo_artworks.csv"
+image_folder = base_path / "demo_images"
+model_file = str(model_file)
+image_csv = str(image_csv)
+image_folder = str(image_folder)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 class_index = {"Drawing": 0, "Photograph": 1, "Print": 2}