Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -258,13 +258,15 @@ with gr.Blocks() as demo:
|
|
258 |
instructions = """## Instructions:
|
259 |
1. Click **Get Random Image** to grab a random image from dataset
|
260 |
1. Click **Create Caption** to generate a caption for the image
|
261 |
-
1. Different models can be selected:
|
|
|
|
|
262 |
1. The dataset (<a href="https://github.com/unsplash/datasets" target="_blank">Unsplash Lite</a>) contains 25,000 nature-focused images"""
|
263 |
gr.Markdown(instructions)
|
264 |
with gr.Row():
|
265 |
with gr.Column(variant="panel"):
|
266 |
im_cap = gr.Image(interactive=False).style(height=height)
|
267 |
-
model_name = gr.Radio(choices=["COCO","Conceptual
|
268 |
with gr.Row():
|
269 |
get_btn_cap = gr.Button("Get Random Image").style(full_width=False)
|
270 |
caption_btn = gr.Button("Create Caption").style(full_width=False)
|
|
|
258 |
instructions = """## Instructions:
|
259 |
1. Click **Get Random Image** to grab a random image from dataset
|
260 |
1. Click **Create Caption** to generate a caption for the image
|
261 |
+
1. Different models can be selected:
|
262 |
+
* **COCO** generally produces more straight-forward captions, but it is a smaller dataset and therefore struggles to recognize certain objects
|
263 |
+
* **Conceptual Captions** is a much larger dataset but sometimes produces overly ``poetic'' results
|
264 |
1. The dataset (<a href="https://github.com/unsplash/datasets" target="_blank">Unsplash Lite</a>) contains 25,000 nature-focused images"""
|
265 |
gr.Markdown(instructions)
|
266 |
with gr.Row():
|
267 |
with gr.Column(variant="panel"):
|
268 |
im_cap = gr.Image(interactive=False).style(height=height)
|
269 |
+
model_name = gr.Radio(choices=["COCO","Conceptual Captions"], type="value", value="COCO", label="Model").style(container=True, item_container = False)
|
270 |
with gr.Row():
|
271 |
get_btn_cap = gr.Button("Get Random Image").style(full_width=False)
|
272 |
caption_btn = gr.Button("Create Caption").style(full_width=False)
|