Update app.py

app.py

@@ -343,16 +343,7 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
     user_repo_id = f"{profile.username}/{slug_name}"
     create_readme(info, downloaded_files, user_repo_id, link_civit, folder=folder)
     try:
-        print("Civit URL: ", url)
-        print("Downloaded files: ", downloaded_files)
-        print("Files in folder")
-        files = [f for f in os.listdir(folder) if os.path.isfile(os.path.join(folder, f))]
-        print(files)
-        print("Repo ID: ", user_repo_id)
-        print("Current folder: ", folder)
-        print(oauth_token.token)
-        print("Creating repo...")
-        print(user_repo_id)
+
         create_repo(repo_id=user_repo_id, private=True, exist_ok=True, token=oauth_token.token)
         print("Uploading repo...")
         upload_folder(
@@ -366,28 +357,6 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
         print(e)
         raise gr.Error("Your Hugging Face Token expired. Log out and in again to upload your models.")
         
-    #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)
-
     return f'''# Model uploaded to 🤗!
     ## Access it here [{user_repo_id}](https://huggingface.co/{user_repo_id}) '''
 
@@ -400,6 +369,7 @@ def bulk_upload(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken,
             try:
                 upload_result = upload_civit_to_hf(profile, oauth_token, url, link_civit)
                 gr.Warning(f"Model {link_civit} uploaded")
+                yield upload_result
             except Exception as e:
                 print(e)
                 gr.Warning(f"Error uploading the model {link_civit}")