Update app.py

app.py

@@ -130,19 +130,34 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
         api = HfApi(token=os.environ["HUGGING_FACE_HUB_TOKEN"])
         username = api.whoami()["name"]
         slug_name = slugify(info["name"])
-        repo_id = f"{username}/{slug_name}"
-        api.create_repo(repo_id=repo_id, private=True, exist_ok=True)
+        repo_id = f"{username}/{profile.preferred_username}-{slug_name}"
+        api.create_repo(repo_id=repo_id, exist_ok=True)
         api.upload_folder(
             folder_path=folder,
             repo_id=repo_id,
             repo_type="model"
         )
-        transfer_repos = gr.load("multimodalart/transfer_repos", api_key=os.environ["HUGGING_FACE_HUB_TOKEN"])
-        transfer_repos(repo_id, f"{profile.preferred_username}/{slug_name}")
+        transfer_repos = gr.load("multimodalart/transfer_repos", api_key=os.environ["HUGGING_FACE_HUB_TOKEN"], src="spaces")
+        user_repo_id = f"{profile.preferred_username}/{slug_name}"
+        response_code = transfer_repos(repo_id, user_repo_id)
+        i = 0
+        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:
+                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:
+                raise gr.Error("It seems this model has been uploaded already in your account.")
+            else:
+                raise gr.Error(f"Unexpected response code: {response_code}.")
+
+            
     except:
         raise gr.Error("something went wrong")
     return f'''### Model uploaded!
-    Access it here [{profile.preferred_username}/{slug_name}](https://huggingface.co/{profile.preferred_username}/{slug_name}) '''
+    Access it here [{user_repo_id}](https://huggingface.co/{user_repo_id}) '''
 
 
 def get_creator(username):