jeff-RQ
/

new-test-model

image-captioning

visual-question-answering

Model card Files Files and versions

jeff-RQ commited on Jul 4, 2023

Commit

1e131a4

·

1 Parent(s): 40bec71

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -21,7 +21,7 @@ class EndpointHandler:
         image_string = base64.b64decode(data["image"])
         image = Image.open(io.BytesIO(image_string))
-        inputs = self.processor(images=image, text=prompt, return_tensors="pt").to(self.device)
         generated_ids = self.model.generate(**inputs)
         generated_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()

         image_string = base64.b64decode(data["image"])
         image = Image.open(io.BytesIO(image_string))
+        inputs = self.processor(images=image, text=text, return_tensors="pt").to(self.device)
         generated_ids = self.model.generate(**inputs)
         generated_text = self.processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()