Update handler.py

handler.py

@@ -19,10 +19,13 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]):
         parameters = data.pop("inputs", data)
-        url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+        url = ["http://images.cocodataset.org/val2017/000000039769.jpg", "http://images.cocodataset.org/val2017/000000039769.jpg","http://images.cocodataset.org/val2017/000000039769.jpg"]
         prompt = "USER: <image>\nWhat are these?\nASSISTANT:"
         raw_image = Image.open(requests.get(url, stream=True).raw)
-        inputs = self.processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
-        output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
-        readable = (self.processor.decode(output[0][2:], skip_special_tokens=True))
-        return readable
+        outputs = []
+        for link in url:
+            inputs = self.processor(prompt, raw_image, return_tensors='pt').to(0, torch.float16)
+            output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
+            print(self.processor.decode(output[0][2:], skip_special_tokens=True))
+            outputs.append(url)
+        return outputs