Update handler.py

handler.py

@@ -42,11 +42,11 @@ class EndpointHandler():
         image_io = io.BytesIO(image_bytes)
         image = Image.open(image_io)
 
-        inputs = processor(images=image, return_tensors="pt").to(device, torch.float16)
+        inputs = self.processor(images=image, return_tensors="pt").to(device, torch.float16)
         pixel_values = inputs.pixel_values
         
-        generated_ids = model.generate(pixel_values=pixel_values, max_length=25)
-        generated_caption = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+        generated_ids = self.model.generate(pixel_values=pixel_values, max_length=25)
+        generated_caption = self.processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
         print(generated_caption)
  
         # raw_images = [Image.open(BytesIO(_img)) for _img in inputs]