Update app.py

app.py

@@ -627,14 +627,17 @@ def upload_to_huggingface(model_path, hf_token, orgs_name, model_name, make_priv
         if not os.path.exists(model_path):
             raise ValueError("Model path does not exist.")
         
-        # Upload model
-        repo_id = f"{orgs_name}/{model_name}" if orgs_name else model_name
+        # Check if repo already exists
         api = HfApi()
-        user = api.whoami(hf_token)
-        if make_private:
-            api.create_repo(repo_id, private=True)
-        else:
-            api.create_repo(repo_id)
+        repo_id = f"{orgs_name}/{model_name}" if orgs_name else model_name
+        try:
+            api.repo_info(repo_id)
+            print(f"⚠️ Repository '{repo_id}' already exists. Proceeding with upload.")
+        except Exception:
+            if make_private:
+                api.create_repo(repo_id, private=True)
+            else:
+                api.create_repo(repo_id)
         
         # Push model files
         api.upload_folder(
@@ -645,9 +648,8 @@ def upload_to_huggingface(model_path, hf_token, orgs_name, model_name, make_priv
             ignore_patterns=".ipynb_checkpoints",
         )
         
-        with output_widget:
-            print(f"Model uploaded to: https://huggingface.co/{repo_id}")
-            return f"Model uploaded to: https://huggingface.co/{repo_id}"
+        print(f"Model uploaded to: https://huggingface.co/{repo_id}")
+        return f"Model uploaded to: https://huggingface.co/{repo_id}"
     except Exception as e:
         error_msg = f"❌ Error during upload: {str(e)}"
         print(error_msg)