Update app.py

app.py

@@ -19,9 +19,14 @@ def download_github_repo(repo_url):
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# Function to download a Hugging Face Space or model/dataset with authentication
-def download_huggingface_repo(repo_id, hf_token):
+# Function to download a Hugging Face Space or model/dataset using an environment variable for the token
+def download_huggingface_repo(repo_id):
     try:
+        # Get Hugging Face token from environment variable
+        hf_token = os.getenv("HF_TOKEN")
+        if hf_token is None:
+            return "Hugging Face token not found. Please set the HF_TOKEN environment variable."
+
         # Log in using the Hugging Face token
         login(token=hf_token)
         
@@ -37,11 +42,11 @@ def download_huggingface_repo(repo_id, hf_token):
         return f"An error occurred: {str(e)}"
 
 # Gradio Interface
-def download_repo(repo_type, repo_url_or_id, hf_token=""):
+def download_repo(repo_type, repo_url_or_id):
     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, hf_token)
+        return download_huggingface_repo(repo_url_or_id)
     else:
         return "Invalid repository type selected."
 
@@ -50,12 +55,11 @@ 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="Hugging Face Token (Leave blank if accessing public repo)", type="password")
+        gr.Textbox(label="Repository URL (GitHub) or Repo ID (Hugging Face)")
     ],
     outputs="text",
     title="Repository Downloader",
-    description="Enter the GitHub repo URL or Hugging Face repo ID to download. If using Hugging Face private repo, provide your access token."
+    description="Enter the GitHub repo URL or Hugging Face repo ID to download."
 )
 
 # Launch the app