Update app.py

app.py

@@ -56,7 +56,8 @@ def extract_info(json_data):
         for model_version in json_data["modelVersions"]:
             if model_version["baseModel"] in ["SDXL 1.0", "SDXL 0.9", "SD 1.5", "SD 1.4", "SD 2.1", "SD 2.0", "SD 2.0 768", "SD 2.1 768"]:
                 for file in model_version["files"]:
-                    if file["primary"]:
+                    print(file)
+                    if "primary" in file:
                         # Start by adding the primary file to the list
                         urls_to_download = [{"url": file["downloadUrl"], "filename": file["name"], "type": "weightName"}]