Update app.py

app.py

@@ -1011,21 +1011,18 @@ imagenet_classes = {
     999: "toilet tissue, toilet paper, bathroom tissue",
 }
 
-path_ffhq = hf_hub_download(repo_id="brownvc/BaselineGAN-FFHQ-256x256", filename="network-snapshot-final.pkl", local_dir="ffhq-256x256.pkl")
-path_imgnet = hf_hub_download(repo_id="brownvc/BaselineGAN-ImgNet-64x64-v0", filename="network-snapshot-final.pkl", local_dir="ImgNet-64x64-v0.pkl")
-
-print(path_ffhq)
-print(path_imgnet)
+path_ffhq = hf_hub_download(repo_id="brownvc/BaselineGAN-FFHQ-256x256", filename="network-snapshot-final.pkl", local_dir="ffhq")
+path_imagenet = hf_hub_download(repo_id="brownvc/BaselineGAN-ImgNet-64x64-v0", filename="network-snapshot-final.pkl", local_dir="imagenet")
 
 # Configuration
 networks = {
     "FFHQ (Faces) 256px": {
-        "path": "ffhq-256x256.pkl",
+        "path": "path_ffhq",
         "G": None,
         "conditional": False
     },
     "ImageNet 64px": {
-        "path": "ImgNet-64x64-v0.pkl",
+        "path": "path_imagenet",
         "G": None,
         "conditional": True
     }