Updating the code for the Gemma 3n trial

src/streamlit_app.py

@@ -3,11 +3,27 @@ from transformers import pipeline
 from PIL import Image
 import torch
 import os
+from huggingface_hub import login
 
 # Set cache directory to avoid permission issues
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
 os.environ["HF_HOME"] = "/tmp/hf_home"
 
+# Authenticate with HuggingFace
+def authenticate_hf():
+    """Authenticate with HuggingFace using token from secrets"""
+    hf_token = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
+    if hf_token:
+        try:
+            login(token=hf_token)
+            return True
+        except Exception as e:
+            st.error(f"Failed to authenticate with HuggingFace: {str(e)}")
+            return False
+    else:
+        st.error("HuggingFace token not found in environment variables")
+        return False
+
 # Set page config
 st.set_page_config(
     page_title="Gemma-3n E4B Vision-Language Model",
@@ -65,18 +81,30 @@ def main():
     st.title("🤖 Gemma-3n E4B Vision-Language Model")
     st.markdown("Upload an image and ask questions about it!")
     
+    # Check authentication first
+    if not authenticate_hf():
+        st.error("❌ Authentication failed. Please check your HuggingFace token in the Space secrets.")
+        st.markdown("""
+        **To fix this:**
+        1. Go to your Space settings (⚙️ icon)
+        2. Navigate to "Repository secrets"
+        3. Add a secret with name: `HF_TOKEN`
+        4. Value: Your HuggingFace token
+        5. Restart the Space
+        """)
+        return
+    
+    st.success("✅ Successfully authenticated with HuggingFace!")
+    
     # Check if user is authenticated
     st.sidebar.markdown("### 📋 Setup Instructions")
     st.sidebar.markdown("""
-    1. Make sure you have access to the gated model
-    2. Login to HuggingFace using your token:
-       ```bash
-       huggingface-cli login
-       ```
-    3. Or set your token as an environment variable:
-       ```bash
-       export HUGGINGFACE_HUB_TOKEN=your_token_here
-       ```
+    ✅ **Authentication**: Configured via Space secrets
+    
+    Make sure you have:
+    1. Access to the gated model
+    2. Added your HF token to Space secrets
+    3. Token has proper permissions
     """)
     
     # Load model