Update app.py

app.py

@@ -13,7 +13,7 @@ def initialize_pipeline():
     return pipeline(
         "image-text-to-text",
         model=model_id,
-        use_auth_token=HF_TOKEN
+        model_kwargs={"torch_dtype": torch.bfloat16, "use_auth_token": HF_TOKEN}
     )
 
 # Preprocess image function