Thomasboosinger
commited on
Commit
•
c74ba54
1
Parent(s):
2d58eea
Update handler.py
Browse files- 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 |
|