Update app.py

app.py

@@ -78,7 +78,7 @@ def infer(audio_file):
     images = pipe(prompt=result).images[0]
     
     #return cap_result, result, images
-    return images, gr.update(visible=True)
+    return images, result, gr.update(visible=True)
 
 css = """
 #col-container {max-width: 510px; margin-left: auto; margin-right: auto;}
@@ -113,7 +113,7 @@ with gr.Blocks(css=css) as demo:
         tryagain_btn = gr.Button("Try again ?", visible=False)
 
     #infer_btn.click(fn=infer, inputs=[audio_input], outputs=[lpmc_cap, llama_trans_cap, img_result])
-    infer_btn.click(fn=infer, inputs=[audio_input], outputs=[img_result, tryagain_btn])
+    infer_btn.click(fn=infer, inputs=[audio_input], outputs=[img_result, llama_trans_cap, tryagain_btn])
     tryagain_btn.click(fn=solo_xd, inputs=[llama_trans_cap], outputs=[img_result])
 
 demo.queue().launch()