Update handler.py

handler.py

@@ -43,7 +43,7 @@ def download_image(url: str) -> Image.Image:
 	image = Image.open(BytesIO(response.content)).convert('RGB')
 	return image
 
-def classify_image(model, image_processor, class_info, device, accuracy):
+def classify_image(model, image_processor, class_info, device, image_url, accuracy):
 
 	# Descargar y procesar imagen
 	image = download_image(image_url)
@@ -86,7 +86,7 @@ class EndpointHandler():
 
 	def predict_objects(self, image_url, accuracy):
 		
-		result_df =  classify_image(image_url, accuracy)   
+		result_df =  classify_image(self.model, self.image_processor, self.class_info, self.device, image_url, accuracy)   
 		return result_df
 
 	def __call__(self, event):