Update app.py

app.py

@@ -13,7 +13,7 @@ print(f"[INFO] Loading Original Model")
 original_pipeline = pipeline(
     "image-text-to-text",
     model=BASE_MODEL_ID,
-    dtype=torch.float32,
+    dtype=torch.bfloat16,
     device_map="cpu"
 )
 
@@ -22,7 +22,7 @@ print(f"[INFO] Loading Fine-tuned Model")
 ft_pipe = pipeline(
     "image-text-to-text",
     model=FINE_TUNED_MODEL_ID,
-    dtype=torch.float32,
+    dtype=torch.bfloat16,
     device_map="cpu"
 )