Update app.py

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 import tempfile
 import os
 from git import Repo
-from huggingface_hub import HfApi, create_repo, upload_folder
-from huggingface_hub import RepositoryNotFoundError
+from huggingface_hub import HfApi, create_repo, upload_folder, HfHubHTTPError
 api = HfApi()
 
 
@@ -35,10 +34,13 @@ def process(git_repo_url, space_destination, user_token):
         )
         return "Done"
         
-    except RepositoryNotFoundError as e:
-        print("HELLO Error !!!")
-    
-        return e
+    except HfHubHTTPError as e:
+        # Check if the error is a "Repository Not Found" error
+        if "401 Client Error" in str(e) and "Repository Not Found" in str(e):
+            print("Repository not found. Creating the repository...")
+        else:
+            # Re-raise the exception if it's not a "Repository Not Found" error
+            raise e
 
 css = """
 div#col-container{