Update app.py

app.py

@@ -83,7 +83,6 @@ def extract_info(json_data):
                             "SD 2.0 768": "stabilityai/stable-diffusion-2"
                         }
                         base_model = model_mapping[model_version["baseModel"]]
-                        print(json_data)
                         info = {
                             "urls_to_download": urls_to_download,
                             "id": model_version["id"],
@@ -329,19 +328,20 @@ def list_civit_models(username):
 def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], oauth_token: gr.OAuthToken, url, link_civit=False):
     if not profile.name:
         return gr.Error("Are you sure you are logged in?")
+
+    gr.Info(f"Starting uploading {url}")
     
     folder = str(uuid.uuid4())
     os.makedirs(folder, exist_ok=False)
     info, downloaded_files = process_url(url, profile, folder=folder)
-    
+    gr.Info(f"Model {info['name']} found, starting download...")
     username = {profile.username}
     slug_name = slugify(info["name"])
     user_repo_id = f"{profile.username}/{slug_name}"
     create_readme(info, downloaded_files, user_repo_id, link_civit, folder=folder)
     try:
-
         create_repo(repo_id=user_repo_id, private=True, exist_ok=True, token=oauth_token.token)
-        print("Uploading repo...")
+        gr.Info("Starting to upload repo to Hugging Face...")
         upload_folder(
             folder_path=folder,
             repo_id=user_repo_id,