Update app.py

app.py

@@ -1,7 +1,7 @@
 import requests
 import os
 import gradio as gr
-from huggingface_hub import HfApi, update_repo_visibility
+from huggingface_hub import HfApi, update_repo_visibility, whoami, upload_folder, create_repo
 from slugify import slugify
 import gradio as gr
 import re
@@ -31,7 +31,6 @@ def check_nsfw(json_data, profile):
         return True
     for model_version in json_data["modelVersions"]:
         for image in model_version["images"]:
-            print(image)
             if image["nsfw"] > 2:
                 return False
     return True
@@ -299,9 +298,10 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
     os.makedirs(folder, exist_ok=False)
     info, downloaded_files = process_url(url, profile, folder=folder)
     try:
-        print(oauth_token.token)
-        api = HfApi(token=oauth_token.token)
-        username = api.whoami()["name"]
+        #print(oauth_token)
+        #print(oauth_token.token)
+        #api = HfApi(token=oauth_token.token)
+        username = whoami(oauth_token.token)["name"]
         slug_name = slugify(info["name"])
     except:
         raise gr.Error("logging into hf went wrong")
@@ -311,9 +311,9 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuth
         #repo_id = f"{username}/{profile.preferred_username}-{slug_name}"
         print("Creating repo...")
         print(user_repo_id)
-        api.create_repo(repo_id=user_repo_id, private=False, exist_ok=True)
+        create_repo(repo_id=user_repo_id, private=False, exist_ok=True)
         print("Uploading repo...")
-        api.upload_folder(
+        upload_folder(
             folder_path=folder,
             repo_id=user_repo_id,
             repo_type="model",