kartikay24 commited on
Commit
40585ee
·
1 Parent(s): 375214f

added test images

Browse files

added test images and acknowledgment given to the actual owner of the model

Files changed (1) hide show
  1. app.py +2 -1
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()