Update app.py

app.py

@@ -126,6 +126,7 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
     os.makedirs(folder, exist_ok=False)
     info, downloaded_files = process_url(url, folder)
     create_readme(info, downloaded_files, folder)
+    print(os.listdir(folder))
     try:
         api = HfApi(token=os.environ["HUGGING_FACE_HUB_TOKEN"])
         username = api.whoami()["name"]
@@ -142,14 +143,14 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
         response_code = transfer_repos(repo_id, user_repo_id)
         print("response_code", response_code)
         i = 0
-        while response_code != 200:
-            if response_code == 404:
+        while response_code != "200":
+            if response_code == "404":
                 raise gr.Error("Something went wrong with the model upload. Try again.")
-            elif response_code == 409 and i < 3:
+            elif response_code == "409" and i < 3:
                 user_repo_id = f"{profile.preferred_username}/{slug_name}-{i}"
                 response_code = transfer_repos(repo_id, user_repo_id)
                 i += 1
-            elif response_code == 409 and i >= 3:
+            elif response_code == "409" and i >= 3:
                 raise gr.Error("It seems this model has been uploaded already in your account.")
             else:
                 raise gr.Error(f"Unexpected response code: {response_code}.")