Add user input
Browse files- handler.py +1 -2
handler.py
CHANGED
@@ -41,10 +41,9 @@ class EndpointHandler:
|
|
41 |
)
|
42 |
self.logger.info("decoded")"""
|
43 |
|
44 |
-
url_1 = "http://images.cocodataset.org/val2017/000000039769.jpg"
|
45 |
url_2 = "http://images.cocodataset.org/val2017/000000219578.jpg"
|
46 |
|
47 |
-
image_1 =
|
48 |
image_2 = Image.open(requests.get(url_2, stream=True).raw)
|
49 |
images = [image_1, image_2]
|
50 |
|
|
|
41 |
)
|
42 |
self.logger.info("decoded")"""
|
43 |
|
|
|
44 |
url_2 = "http://images.cocodataset.org/val2017/000000219578.jpg"
|
45 |
|
46 |
+
image_1 = data.pop("inputs", data)
|
47 |
image_2 = Image.open(requests.get(url_2, stream=True).raw)
|
48 |
images = [image_1, image_2]
|
49 |
|