Update app.py

app.py

@@ -7,11 +7,11 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 model = torch.hub.load('mair-lab/mapl', 'mapl')
 model.eval()
-model.to(device, torch.bfloat16)
+model.to(device, torch.float16)
 
 
 def predict(image: Image.Image, question: str) -> str:
-    pixel_values = model.image_transform(image).unsqueeze(0).to(device, torch.bfloat16)
+    pixel_values = model.image_transform(image).unsqueeze(0).to(device, torch.float16)
 
     input_ids = None
     if question: