Spaces:
Runtime error
Runtime error
Commit
·
40585ee
1
Parent(s):
375214f
added test images
Browse filesadded test images and acknowledgment given to the actual owner of the model
app.py
CHANGED
@@ -32,5 +32,6 @@ input_image = gr.inputs.Image(label="Upload an Image")
|
|
32 |
input_text = gr.inputs.Textbox(label="Enter Text (for Conditional Captioning)")
|
33 |
radio_button = gr.inputs.Radio(choices=["Conditional", "Unconditional"], label="Captioning Type")
|
34 |
output_text = gr.outputs.Textbox(label="Caption")
|
|
|
35 |
|
36 |
-
gr.Interface(fn=generate_caption, inputs=[input_image, radio_button, input_text], outputs=output_text, title="Image Captioning").launch()
|
|
|
32 |
input_text = gr.inputs.Textbox(label="Enter Text (for Conditional Captioning)")
|
33 |
radio_button = gr.inputs.Radio(choices=["Conditional", "Unconditional"], label="Captioning Type")
|
34 |
output_text = gr.outputs.Textbox(label="Caption")
|
35 |
+
examples = [f"Image{i}.png" for i in range(1,4)]
|
36 |
|
37 |
+
gr.Interface(fn=generate_caption, inputs=[input_image, radio_button, input_text], outputs=output_text, examples = examples, title="Image Captioning", description = "Model was taken from https://huggingface.co/Salesforce/blip-image-captioning-base").launch()
|