EntrepreneurFirst
/

idefics-deployment-fonction

Inference Endpoints

Model card Files Files and versions Community

mcordier95 commited on Apr 25

Commit

9f72b3f

•

1 Parent(s): f9fb547

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -30,7 +30,7 @@ class EndpointHandler():
             }
         """
         print("4")
-        text = self.processor.apply_chat_template(data["inputs"]["messages"], add_generation_prompt=True)
         print("5")
         images = [self.decode_image_base64(img) for img in data["inputs"]["images"]]
         print("6")

             }
         """
         print("4")
+        text = self.processor.apply_chat_template(data["inputs"]["messages"], add_generation_prompt=False)
         print("5")
         images = [self.decode_image_base64(img) for img in data["inputs"]["images"]]
         print("6")