Update app.py

app.py

@@ -109,7 +109,7 @@ def infer_and_log(text_input):
     return formatted_output, word_probs
 
 def clear_fields():
-    return "", ""
+    return "", "", ""
 setup_hf_dataset()
 
 with gr.Blocks() as app:
@@ -123,7 +123,7 @@ with gr.Blocks() as app:
         submit_btn = gr.Button("Submit")
         clear_btn = gr.Button("Clear")
     submit_btn.click(fn=infer_and_log, inputs=input_box, outputs=[output_html, output_json])
-    clear_btn.click(fn=clear_fields, outputs=[input_box, output_box])
+    clear_btn.click(fn=clear_fields, outputs=[input_box, output_html, output_json])
     
 if __name__ == "__main__":
     app.launch()