lyttt commited on
Commit
cfc2299
·
1 Parent(s): 8132819
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def greet(image):
12
  image = (np.clip(image, 0, 1) * 255).astype(np.uint8)
13
  image = Image.fromarray(image, mode='RGB')
14
  with torch.no_grad():
15
- outputs = model([image]).generated_text[0]
16
  return outputs
17
 
18
  demo = gr.Interface(fn=greet, inputs="image", outputs="text")
 
12
  image = (np.clip(image, 0, 1) * 255).astype(np.uint8)
13
  image = Image.fromarray(image, mode='RGB')
14
  with torch.no_grad():
15
+ outputs = model([image], 300).generated_text[0]
16
  return outputs
17
 
18
  demo = gr.Interface(fn=greet, inputs="image", outputs="text")