Update app.py

app.py

@@ -1011,12 +1011,15 @@ imagenet_classes = {
     999: "toilet tissue, toilet paper, bathroom tissue",
 }
 
-hf_hub_download(repo_id="brownvc/BaselineGAN-FFHQ-256x256", filename="network-snapshot-final.pkl", local_dir="ffhq-256x256.pkl")
-hf_hub_download(repo_id="brownvc/BaselineGAN-ImgNet-64x64-v0", filename="network-snapshot-final.pkl", local_dir="ImgNet-64x64-v0.pkl")
+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)
 
 # Configuration
 networks = {
-    "FFHQ (Faces) 128px": {
+    "FFHQ (Faces) 256px": {
         "path": "ffhq-256x256.pkl",
         "G": None,
         "conditional": False