Update app.py

app.py

@@ -10,10 +10,9 @@ import time
 print("Loading MedGemma model...")
 pipe = pipeline(
     "image-text-to-text",
-    model="unsloth/medgemma-4b-it-bnb-4bit",
+    model="google/medgemma-4b-it",
     torch_dtype=torch.bfloat16,
-    # device="cuda" if torch.cuda.is_available() else "cpu",
-    device_map="auto",
+    device="cuda" if torch.cuda.is_available() else "cpu",
 )
 print("Model loaded successfully!")