Update app.py

app.py

@@ -48,7 +48,7 @@ def chat_with_image(image, user_message):
         new_inputs = {}
         for k, v in prepare_inputs.items():
             if torch.is_tensor(v):
-                if k in ["input_ids", "labels"]:
+                if k in ["input_ids", "labels","attention_mask"]:
                     new_inputs[k] = v.to(torch.long)
                 else:
                     new_inputs[k] = v.to(torch.float16)