add progress bar

app.py

@@ -48,7 +48,7 @@ def export_to_video(frames: np.ndarray, fps: int) -> str:
     writer.close()
     return out_file.name
 
-def infer(image_init):
+def infer(image_init, progress=gr.Progress(track_tqdm=True)):
     prompt = get_caption(image_init)
     video_frames = pipe(prompt, num_inference_steps=40, height=320, width=576, num_frames=24).frames[0]
     video_path = export_to_video(video_frames, 12)
@@ -138,7 +138,7 @@ with gr.Blocks(css=css) as demo:
         image_init = gr.Image(label="Image Init", type="filepath", sources=["upload"], elem_id="image-init")
         #inference_steps = gr.Slider(label="Inference Steps", minimum=10, maximum=100, step=1, value=40, interactive=False)
         submit_btn = gr.Button("Submit")
-        coca_cap = gr.Textbox(label="Caption", placeholder="CoCa Caption will be displayed here", elem_id="coca-cap-in")
+        coca_cap = gr.Textbox(label="Caption", placeholder="Kosmos-2 caption will be displayed here", elem_id="coca-cap-in")
         video_result = gr.Video(label="Video Output", elem_id="video-output")
 
     submit_btn.click(