Update app.py

app.py

@@ -36,7 +36,7 @@ def identity(index, split):
     return ds["prompt"], ds["completion"]
     
 def save(index, language, split, prompt, completion):
-    with open("./output.jsonl", "a") as fout :
+    with open("/home/user/app/output.jsonl", "a") as fout :
         fout.write(
             json.dumps(
                 {
@@ -48,7 +48,7 @@ def save(index, language, split, prompt, completion):
             )+"\n"
         )
     next_index = min(1+index, len(samples[split])-1)
-    return next_index, samples[split][next_index]["prompt"], samples[split][next_index]["completion"]
+    return next_index, samples[split][next_index]["prompt"], samples[split][next_index]["completion"], "", ""
 
 with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
     gr.HTML("""<h1 align="center">MT💫</h1>""")
@@ -76,6 +76,6 @@ with gr.Blocks(analytics_enabled=False, css=custom_css) as demo:
     
     split.change(caller_split, inputs=[split], outputs=[index_example, prompt, completion])
     index_example.release(identity, inputs=[index_example, split], outputs=[prompt, completion])
-    button.click(save, inputs=[index_example, language, split, translated_prompt, translated_completion], outputs=[index_example, prompt, completion])
+    button.click(save, inputs=[index_example, language, split, translated_prompt, translated_completion], outputs=[index_example, prompt, completion, translated_prompt, translated_completion])
     
 demo.launch(debug=True)