Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 import git
-from huggingface_hub import snapshot_download
+from huggingface_hub import snapshot_download, login
 import shutil
 
 # Function to clone a GitHub repository
@@ -19,21 +19,24 @@ 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
-def download_huggingface_repo(repo_id):
+# Function to download a Hugging Face model or 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)
+        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)}"
 
 # Gradio Interface
-def download_repo(repo_type, repo_url_or_id):
+def download_repo(repo_type, repo_url_or_id, hf_token=""):
     if repo_type == "GitHub":
         return download_github_repo(repo_url_or_id)
     elif repo_type == "Hugging Face":
-        return download_huggingface_repo(repo_url_or_id)
+        return download_huggingface_repo(repo_url_or_id, hf_token)
     else:
         return "Invalid repository type selected."
 
@@ -42,11 +45,12 @@ interface = gr.Interface(
     fn=download_repo,
     inputs=[
         gr.Radio(choices=["GitHub", "Hugging Face"], label="Repository Type"),
-        gr.Textbox(label="Repository URL (GitHub) or Repo ID (Hugging Face)")
+        gr.Textbox(label="Repository URL (GitHub) or Repo ID (Hugging Face)"),
+        gr.Textbox(label="Hugging Face Token (Leave blank if accessing public repo)", type="password")
     ],
     outputs="text",
     title="Repository Downloader",
-    description="Enter the GitHub repo URL or Hugging Face repo ID to download."
+    description="Enter the GitHub repo URL or Hugging Face repo ID to download. If using Hugging Face private repo, provide your access token."
 )
 
 # Launch the app