Update handler.py
Browse files- handler.py +1 -3
handler.py
CHANGED
@@ -33,10 +33,8 @@ class EndpointHandler():
|
|
33 |
"""
|
34 |
inputs = data.pop("inputs", data)
|
35 |
parameters = data.pop("parameters", {})
|
36 |
-
|
37 |
-
raw_images = [Image.open(BytesIO(_img)) for _img in inputs]
|
38 |
|
39 |
-
processed_image = self.processor(images=
|
40 |
processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
|
41 |
processed_image = {**processed_image, **parameters}
|
42 |
|
|
|
33 |
"""
|
34 |
inputs = data.pop("inputs", data)
|
35 |
parameters = data.pop("parameters", {})
|
|
|
|
|
36 |
|
37 |
+
processed_image = self.processor(images=inputs, return_tensors="pt")
|
38 |
processed_image["pixel_values"] = processed_image["pixel_values"].to(device)
|
39 |
processed_image = {**processed_image, **parameters}
|
40 |
|