Update app.py

app.py

@@ -30,8 +30,8 @@ def set_labels(text):
     return text.split(",")
     
 get_caption = gr.load("ryaalbr/caption", src="spaces", hf_token=environ["api_key"])
-def generate_text(image):
-    return get_caption(image)
+def generate_text(image, model_name):
+    return get_caption(image, model_name)
     
 with gr.Blocks() as demo:
 
@@ -68,14 +68,15 @@ with gr.Blocks() as demo:
     with gr.Tab("Image Captioning"):
         with gr.Row():
             with gr.Column(variant="panel"):
-                im_cap = gr.Image(interactive=False).style(height=height)
+                im_cap = gr.Image(interactive=False, type='filepath').style(height=height)
+                model_name = gr.Radio(choices=["COCO","Conceptual captions"], type="value", value="COCO", label="Model")
                 with gr.Row():
                     get_btn_cap = gr.Button("Get Random Image").style(full_width=False)
                     caption_btn = gr.Button("Create Caption").style(full_width=False)
-            caption = gr.Text()
+            caption = gr.Textbox(label='Caption')
         get_btn_cap.click(fn=rand_image, outputs=im_cap)
         #im_cap.change(generate_text, inputs=im_cap, outputs=caption)
-        caption_btn.click(generate_text, inputs=im_cap, outputs=caption)        
+        caption_btn.click(generate_text, inputs=[im_cap, model_name], outputs=caption)        
 
 demo.queue()
 demo.launch()