Update app.py

app.py

@@ -117,10 +117,14 @@ from huggingface_hub import login
 hf_token = os.getenv("hf_token")
 login(token=hf_token)
 
+
+model_id = "google/gemma-3n-E4B-it-litert-preview"
+# model_id = "google/medgemma-4b-it"
+
 # Load the MedGemma pipeline
 pipe = pipeline(
     "image-text-to-text",
-    model="google/medgemma-4b-it",
+    model=model_id,
     torch_dtype=torch.bfloat16,
     device="cuda" if torch.cuda.is_available() else "cpu",
 )