Update app.py

app.py

@@ -32,11 +32,11 @@ def arena_function(input_text):
     output_text1,output_text2 = generate_text(input_text,1),generate_text(input_text,1)
     data_to_store = {
         "input": input_text,
-        "r1": output_text1,
-        "r2": output_text2,
+        "r1": output_text1[0]['generated_text'],
+        "r2": output_text2[0]['generated_text'],
     }
     id=store_in_mongodb("arena_collection", data_to_store)
-    return  output_text1, output_text2, id.inserted_id,btn_enable,btn_enable,btn_enable,btn_enable
+    return  output_text1[0]['generated_text'], output_text2[0]['generated_text'], id.inserted_id,btn_enable,btn_enable,btn_enable,btn_enable
 
 general_interface = gr.Interface(fn=general_function, inputs=gr.Textbox(label="Enter your text here:", min_width=600), outputs="text")