Update app.py

app.py

@@ -66,7 +66,9 @@ with gr.Blocks() as demo:
         return result, gr.update(visible=True), text, pred
 
     predict_btn.click(classify_review, inputs=input_text, outputs=[output_label, hidden_text, hidden_pred])\
-               .then(show_correction_ui, outputs=[output_label, correction_row, hidden_text, hidden_pred])
+           .then(show_correction_ui, inputs=[output_label, hidden_text, hidden_pred],
+                 outputs=[output_label, correction_row, hidden_text, hidden_pred])
+
 
     submit_btn.click(save_correction, inputs=[hidden_text, hidden_pred, correction_dropdown], outputs=correction_status)