Update handler.py

handler.py

@@ -16,8 +16,8 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
 
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
-        inputs = self.processor(image, return_tensors="pt").to(device)
-        outputs = self.model(**inputs)
+        inputs = self.processor(image, inputs['text'], return_tensors="pt").to(device)
+        outputs = self.model.generate(**inputs)
         
-        return self.processor.decode(outputs[0], skip_special_tokens=True)
+        return self.processor.decode(out[0], skip_special_tokens=True)