Update handler.py

handler.py

@@ -21,38 +21,18 @@ class EndpointHandler():
         parameters = data.pop("inputs", data)
         url = ["http://images.cocodataset.org/val2017/000000039769.jpg", 
                "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg", 
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg", 
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg", 
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-               "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg", 
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg", 
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg", 
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
-                 "http://images.cocodataset.org/val2017/000000039769.jpg",
+                "http://images.cocodataset.org/val2017/000000039769.jpg",
+                "http://images.cocodataset.org/val2017/000000039769.jpg",
                 "http://images.cocodataset.org/val2017/000000039769.jpg",
                 "http://images.cocodataset.org/val2017/000000039769.jpg",
                 "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)
+    
         outputs = []
         for link in url:
+            raw_image = Image.open(requests.get(link, 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)
             print(self.processor.decode(output[0][2:], skip_special_tokens=True))