Thomasboosinger commited on
Commit
c74ba54
1 Parent(s): 2d58eea

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -0
handler.py CHANGED
@@ -25,6 +25,9 @@ class EndpointHandler():
25
  # Decode the base64 image to a PIL image
26
  image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
27
 
 
 
 
28
  # Correctly passing the image and candidate labels to the pipeline
29
  detection_results = self.pipeline(image=image, candidate_labels=inputs["candidates"])
30
 
 
25
  # Decode the base64 image to a PIL image
26
  image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
27
 
28
+ # Get candidate labels
29
+ candidate_labels=inputs["candidates"]
30
+
31
  # Correctly passing the image and candidate labels to the pipeline
32
  detection_results = self.pipeline(image=image, candidate_labels=inputs["candidates"])
33