Update app.py

app.py

@@ -1,7 +1,7 @@
 import requests
 import os
 import gradio as gr
-from huggingface_hub import update_repo_visibility, whoami, upload_folder, create_repo, upload_file
+from huggingface_hub import update_repo_visibility, whoami, upload_folder, create_repo, upload_file, update_repo_visibility
 from slugify import slugify
 import gradio as gr
 import re
@@ -314,26 +314,15 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
         print(user_repo_id)
         create_repo(repo_id=user_repo_id, private=False, exist_ok=True, token=oauth_token.token)
         print("Uploading repo...")
-        for filename in os.listdir(folder):
-            file_path = os.path.join(folder, filename)
-            print(filename)
-            print("size: ", os.path.getsize(file_path))
-            print(oauth_token.token)
-            upload_file(
-                path_or_fileobj=file_path,
-                path_in_repo=filename,
-                repo_id=user_repo_id,
-                repo_type="model",
-                token=oauth_token.token
-            )
-       # upload_folder(
-       #     folder_path=folder,
-       #     repo_id=user_repo_id,
-       #     repo_type="model",
-       #     token=oauth_token.token
-       # )
         
-        #api.update_repo_visibility(repo_id=user_repo_id, private=False)
+       upload_folder(
+            folder_path=folder,
+            repo_id=user_repo_id,
+            repo_type="model",
+            token=oauth_token.token
+        )
+        
+        update_repo_visibility(repo_id=user_repo_id, private=False)
     except:
         raise gr.Error("uploading the repo went wrong")