Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,4 +26,4 @@ def caption_generator(image, num_captions):
|
|
26 |
captions = captions +processor.decode(caption, skip_special_tokens=True) + " ,"
|
27 |
return captions
|
28 |
|
29 |
-
gr.Interface(caption_generator, inputs= [gr.
|
|
|
26 |
captions = captions +processor.decode(caption, skip_special_tokens=True) + " ,"
|
27 |
return captions
|
28 |
|
29 |
+
gr.Interface(caption_generator, inputs= [gr.Image(), gr.Dropdown([1, 2, 3, 4,5],value = [2], label = "select no.of captions to generate")], outputs = gr.outputs.Textbox(), live = True).launch()
|