update

app.py

@@ -2,23 +2,41 @@ import os
 import gradio as gr
 import warnings
 
-from huggingface_hub import Repository
-from huggingface_hub import login
-
 warnings.filterwarnings("ignore")
 
-login(token = os.environ['HF_TOKEN'])
+# Optional Hugging Face integration
+try:
+    from huggingface_hub import Repository, login
+    
+    # Try to login if token is available
+    if 'HF_TOKEN' in os.environ and os.environ['HF_TOKEN'].strip():
+        login(token=os.environ['HF_TOKEN'])
+        print("✅ Logged into Hugging Face")
+        
+        # Try to pull dataset if specified
+        if 'DATASET' in os.environ and os.environ['DATASET'].strip():
+            try:
+                repo = Repository(
+                    local_dir="",
+                    repo_type="dataset",
+                    clone_from=os.environ['DATASET'],
+                    token=True
+                )
+                repo.git_pull()
+                print("✅ Dataset pulled successfully")
+            except Exception as e:
+                print(f"⚠️ Warning: Could not pull dataset: {e}")
+    else:
+        print("ℹ️ No HF_TOKEN found, skipping Hugging Face authentication")
+        
+except ImportError as e:
+    print(f"⚠️ Warning: Could not import Hugging Face Hub: {e}")
+except Exception as e:
+    print(f"⚠️ Warning: Hugging Face authentication failed: {e}")
 
+# Import the AI detector
 from model import AIContentDetector
 
-repo = Repository(
-    local_dir="",
-    repo_type="dataset",
-    clone_from=os.environ['DATASET'],
-    token=True
-)
-repo.git_pull()
-
 # Initialize the detector
 detector = AIContentDetector()
 

requirements.txt

@@ -14,3 +14,4 @@ boto3>=1.26.0
 pydub>=0.25.0
 psutil>=5.8.0
 pynvml>=11.0.0
+huggingface_hub>=0.15.0