Cristiants commited on
Commit
869cbdf
Β·
1 Parent(s): cfe136c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -66,8 +66,10 @@ def predict(imageurl):
66
  # nucleus sampling
67
  # caption = model.generate(image, sample=True, top_p=0.9, max_length=20, min_length=5)
68
  return('caption: '+caption[0])
 
69
  demo = gr.Interface(fn=predict,
70
- inputs="text",
71
- outputs=gr.outputs.Label(num_top_classes=3)
72
- )
 
73
  demo.launch()
 
66
  # nucleus sampling
67
  # caption = model.generate(image, sample=True, top_p=0.9, max_length=20, min_length=5)
68
  return('caption: '+caption[0])
69
+
70
  demo = gr.Interface(fn=predict,
71
+ inputs="text",
72
+ outputs=gr.outputs.Label(num_top_classes=3)
73
+ )
74
+
75
  demo.launch()