Spaces:

jinlinyi
/

PerspectiveFields

Running on T4

jinlinyi commited on May 10, 2023

Commit

aadf06d

1 Parent(s): 71c19c0

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -167,8 +167,8 @@ model_zoo = {
         'param': True,
     },
 }
-for model_id in model_zoo[model_id]:
-    html = model_zoo[model_id]['weights']
     if not os.path.exists(os.path.join('models', html.split('/')[-1])):
         os.system(f"wget -P models/ {html}")

         'param': True,
     },
 }
+for model_id in model_zoo:
+    html = model_zoo[model_id]['weights'][0]
     if not os.path.exists(os.path.join('models', html.split('/')[-1])):
         os.system(f"wget -P models/ {html}")