Update app.py

app.py

@@ -19,14 +19,19 @@ def download_github_repo(repo_url):
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# Function to download a Hugging Face model or dataset with authentication
+# Function to download a Hugging Face Space or model/dataset with authentication
 def download_huggingface_repo(repo_id, hf_token):
     try:
         # Log in using the Hugging Face token
         login(token=hf_token)
         
-        # Download the Hugging Face model or dataset snapshot
-        repo_path = snapshot_download(repo_id, token=hf_token)
+        # Check if it's a Hugging Face Space
+        if repo_id.startswith("spaces/"):
+            repo_path = snapshot_download(repo_id, repo_type="space", token=hf_token)
+        else:
+            # Download the Hugging Face model or dataset snapshot
+            repo_path = snapshot_download(repo_id, token=hf_token)
+
         return f"Hugging Face repository '{repo_id}' downloaded successfully."
     except Exception as e:
         return f"An error occurred: {str(e)}"