Update app.py

app.py

@@ -289,7 +289,7 @@ def list_civit_models(username):
     
     return urls
 
-def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, link_civit=False, progress=gr.Progress(track_tqdm=True)):
+def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken, url, link_civit=False, progress=gr.Progress(track_tqdm=True)):
     if not profile.name:
         return gr.Error("Are you sure you are logged in?")
     
@@ -305,41 +305,42 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, link_civit=False
     user_repo_id = f"{profile.preferred_username}/{slug_name}"
     create_readme(info, downloaded_files, user_repo_id, link_civit, folder=folder)
     try:
-        repo_id = f"{username}/{profile.preferred_username}-{slug_name}"
+        #repo_id = f"{username}/{profile.preferred_username}-{slug_name}"
         print("Creating repo...")
-        api.create_repo(repo_id=repo_id, private=True, exist_ok=True)
+        api.create_repo(repo_id=user_repo_id, private=True, exist_ok=True)
         print("Uploading repo...")
         api.upload_folder(
             folder_path=folder,
             repo_id=repo_id,
             repo_type="model",
+            token=oauth_token,
         )
         print("Changing repo visibility...")
         api.update_repo_visibility(repo_id=repo_id, private=False)
     except:
         raise gr.Error("uploading the repo went wrong")
         
-    transfer_repos = gr.load("multimodalart/transfer_repos", hf_token=os.environ["HUGGING_FACE_HUB_TOKEN"], src="spaces")
+    #transfer_repos = gr.load("multimodalart/transfer_repos", hf_token=os.environ["HUGGING_FACE_HUB_TOKEN"], src="spaces")
     
-    response_code = transfer_repos(repo_id, user_repo_id)
-    i = 0
-    while response_code != "200":
-        message = None
-        if response_code == "409":
-            if i < 3:
-                user_repo_id = f"{profile.preferred_username}/{slug_name}-{i}"
-                response_code = transfer_repos(repo_id, user_repo_id)
-                i += 1
-            else:
-                message = "It seems this model has been uploaded already in your account."
-        elif response_code == "404":
-            message = "Something went wrong with the model upload. Try again."
-        else:
-            message = f"Unexpected response code: {response_code}."
-    
-        if message:
-            api.delete_repo(repo_id=repo_id, repo_type="model")
-            raise gr.Error(message)
+    #response_code = transfer_repos(repo_id, user_repo_id)
+    #i = 0
+    #while response_code != "200":
+    #    message = None
+    #    if response_code == "409":
+    #        if i < 3:
+    #            user_repo_id = f"{profile.preferred_username}/{slug_name}-{i}"
+    #            response_code = transfer_repos(repo_id, user_repo_id)
+    #            i += 1
+    #        else:
+    #            message = "It seems this model has been uploaded already in your account."
+    #    elif response_code == "404":
+    #        message = "Something went wrong with the model upload. Try again."
+    #    else:
+    #        message = f"Unexpected response code: {response_code}."
+    #
+    #    if message:
+    #        api.delete_repo(repo_id=repo_id, repo_type="model")
+    #        raise gr.Error(message)
 
     return f'''# Model uploaded to 🤗!
     ## Access it here [{user_repo_id}](https://huggingface.co/{user_repo_id}) '''