Upload app.py with huggingface_hub

app.py

@@ -108,10 +108,23 @@ def load_model():
         try:
             # Download model files from Hugging Face
             print("Downloading model files from Hugging Face...")
+            print(f"Repository: {MODEL_REPO_ID}")
+            
+            # Try to download with authentication
+            try:
+                from huggingface_hub import login
+                # Check if token is available via environment variable
+                token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
+                if token:
+                    login(token=token, add_to_git_credential=False)
+            except:
+                pass  # Token might not be set, that's okay if repo is public
+            
             model_dir = snapshot_download(
                 repo_id=MODEL_REPO_ID,
                 cache_dir=None,
-                local_files_only=False
+                local_files_only=False,
+                token=os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
             )
             
             # Find checkpoint and vocab files
@@ -169,11 +182,32 @@ def load_model():
             return "✅ Model loaded successfully!"
             
         except Exception as e:
-            error_msg = f"❌ Error loading model: {str(e)}"
-            print(error_msg)
+            error_msg = str(e)
+            error_details = f"❌ Error loading model: {error_msg}"
+            print(error_details)
+            
+            # Check if it's an authentication error
+            if "401" in error_msg or "Repository Not Found" in error_msg or "Invalid username or password" in error_msg:
+                detailed_error = (
+                    f"❌ Authentication Error: The model repository '{MODEL_REPO_ID}' is private or requires authentication.\n\n"
+                    f"**Solutions:**\n"
+                    f"1. **Make repository public** (Recommended for playground):\n"
+                    f"   - Go to: https://huggingface.co/{MODEL_REPO_ID}/settings\n"
+                    f"   - Change visibility to 'Public'\n\n"
+                    f"2. **Add authentication token to Space** (if keeping private):\n"
+                    f"   - Go to Space Settings → Repository secrets\n"
+                    f"   - Add secret: HF_TOKEN with your Hugging Face token\n"
+                    f"   - Get token from: https://huggingface.co/settings/tokens\n\n"
+                    f"3. **Upload model files directly to Space** (alternative):\n"
+                    f"   - Upload model checkpoint and vocab.txt to the Space repository\n"
+                    f"   - The app will use local files instead"
+                )
+                print(detailed_error)
+                return detailed_error
+            
             import traceback
             traceback.print_exc()
-            return error_msg
+            return error_details
     
     return "✅ Model already loaded!"