Update app.py

app.py

@@ -4,13 +4,23 @@ from PIL import Image
 import requests
 import torch
 import io
+from huggingface_hub import login
+
+
+
+hf_token = os.environ.get("HF_TOKEN")
+login(token=hf_token)
+
 
 # Initialize the model and processor
 model_id = "google/gemma-3n-e4b-it"
-model = Gemma3nForConditionalGeneration.from_pretrained(
-    model_id, device_map="auto", torch_dtype=torch.bfloat16
-).eval()
-processor = AutoProcessor.from_pretrained(model_id)
+try:
+    model = Gemma3nForConditionalGeneration.from_pretrained(
+        model_id, device_map="auto", torch_dtype=torch.bfloat16
+    ).eval()
+    processor = AutoProcessor.from_pretrained(model_id)
+except Exception as e:
+    raise Exception(f"Failed to load model or processor: {str(e)}")
 
 def process_inputs(image_input, image_url, text_prompt):
     """
@@ -88,8 +98,8 @@ iface = gr.Interface(
         gr.Textbox(label="Text Prompt", placeholder="Enter your prompt here")
     ],
     outputs=gr.Textbox(label="Model Response"),
-    title="Gemma-3 Multimodal App",
-    description="Upload an image or provide an image URL, and enter a text prompt to interact with the Gemma-3 model. The model can describe images, answer questions about them, or respond to text-only prompts.",
+    title="Gemma-3 Multimodal App (Authenticated)",
+    description="Upload an image or provide an image URL, and enter a text prompt to interact with the Gemma-3 model. Ensure you have authenticated with a valid Hugging Face access token.",
     allow_flagging="never"
 )