ybelkada commited on
Commit
ec6a005
·
1 Parent(s): c63643f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ def parse_input(input):
19
 
20
  return image
21
 
22
- def launch(image):
 
23
  inputs = processor(image, return_tensors="pt")
24
  out = model.generate(**inputs)
25
  return processor.decode(out[0], skip_special_tokens=True)
 
19
 
20
  return image
21
 
22
+ def launch(input):
23
+ image = parse_input(input)
24
  inputs = processor(image, return_tensors="pt")
25
  out = model.generate(**inputs)
26
  return processor.decode(out[0], skip_special_tokens=True)