Update handler.py

handler.py

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