Update app.py

app.py

@@ -29,14 +29,14 @@ def general_function(input_text):
     return  output_text
 
 def arena_function(input_text):
-    output_text = generate_text(input_text,2)
+    output_text1,output_text2 = generate_text(input_text,1),generate_text(input_text,1)
     data_to_store = {
         "input": input_text,
-        "r1": output_text[0]['generated_text'],
-        "r2": output_text[1]['generated_text'],
+        "r1": output_text1,
+        "r2": output_text2,
     }
     id=store_in_mongodb("arena_collection", data_to_store)
-    return  output_text[0]['generated_text'], output_text[1]['generated_text'], id.inserted_id,btn_enable,btn_enable,btn_enable,btn_enable
+    return  output_text1, output_text2, 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")