Spaces:

RAYAuser
/

fashion-mnist-syntdata-generator.raytech

Sleeping

RAYAuser commited on Sep 1

Commit

622fbfb

verified ·

1 Parent(s): 2b1b4b7

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -96,7 +96,7 @@ Z_DIM = 100
 NGF = 64
 NUM_CLASSES = 10
 IMAGE_SIZE = 32
-MODEL_FILE = "Raygan-zalando_datasetsgen.pth"
 class_names = [
     "T-shirt/top", "Trouser", "Pullover", "Dress", "Coat",

 NGF = 64
 NUM_CLASSES = 10
 IMAGE_SIZE = 32
+MODEL_FILE = os.path.join(os.path.dirname(__file__), "Raygan-zalando_datasetsgen.pth")
 class_names = [
     "T-shirt/top", "Trouser", "Pullover", "Dress", "Coat",