Update app.py

app.py

@@ -1,7 +1,7 @@
 import requests
 import os
 import gradio as gr
-from huggingface_hub import HfApi
+from huggingface_hub import HfApi, update_repo_visibility
 from slugify import slugify
 import gradio as gr
 import uuid
@@ -130,33 +130,40 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
         username = api.whoami()["name"]
         slug_name = slugify(info["name"])
         repo_id = f"{username}/{profile.preferred_username}-{slug_name}"
-        api.create_repo(repo_id=repo_id, exist_ok=True)
+        api.create_repo(repo_id=repo_id, private=True, exist_ok=True)
         api.upload_folder(
             folder_path=folder,
             repo_id=repo_id,
-            repo_type="model"
+            repo_type="model",
         )
-        transfer_repos = gr.load("multimodalart/transfer_repos", hf_token=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:
+        api.update_repo_visibility(repo_id=repo_id, private=False)
+    except:
+        raise gr.Error("something went wrong")
+        
+    transfer_repos = gr.load("multimodalart/transfer_repos", hf_token=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":
+        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
-            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}.")
+                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)
 
-            
-    except:
-        raise gr.Error("something went wrong")
-    return f'''### Model uploaded!
-    Access it here [{user_repo_id}](https://huggingface.co/{user_repo_id}) '''
+    return f'''# Model uploaded to 🤗!
+    ## Access it here [{user_repo_id}](https://huggingface.co/{user_repo_id}) '''
 
 
 def get_creator(username):