Update app.py

app.py

@@ -1,38 +1,19 @@
 import os
 import gradio as gr
 import warnings
+from huggingface_hub import Repository, login
 
 warnings.filterwarnings("ignore")
 
-# 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}")
+login(token=os.environ['HF_TOKEN'])
+
+repo = Repository(
+    local_dir="",
+    repo_type="dataset",
+    clone_from=os.environ['DATASET'],
+    token=True
+)
+repo.git_pull()
 
 # Import the AI detector
 from model import AIContentDetector