Update app.py

app.py

@@ -9,6 +9,7 @@ api = HfApi()
 REPO_TYPES = ["model", "dataset", "space"]
 
 def upload_model(source_url, dst_repo, token,civitai_api_key, new_name, dst_repo_path, repo_type):
+    try:
         _ = whoami(token)  # Check token validity
 
         # Check if destination path ends with '/'
@@ -68,7 +69,8 @@ def upload_model(source_url, dst_repo, token,civitai_api_key, new_name, dst_repo
             f'Your model has been successfully uploaded to <a href="{repo_url}" target="_blank" style="text-decoration:underline">your Hugging Face repository</a>.',
             "success.jpg",
         )
-
+    except Exception as e:
+        return str(e), "error.jpg"
 
 # Interface setup
 interface = gr.Interface(