Update handler.py

handler.py

@@ -19,9 +19,10 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
         if parameters is not None:
             url = "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)
-            prompt = "USER: <image>\nWhat are these?\nASSISTANT:"
+           
             output = self.model.generate(**inputs, max_new_tokens=200, do_sample=False)
         return output