huz-relay commited on
Commit
1b88ea1
1 Parent(s): d29d44e

Add user input

Browse files
Files changed (1) hide show
  1. 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 = Image.open(requests.get(url_1, stream=True).raw)
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