Update app.py

app.py

@@ -42,7 +42,7 @@ def inference(input_sentence, max_length, temperature,top_k, top_p, greedy_decod
         payload
     )
     print(data)
-    return data[0]
+    return data[0]["generated_text"]
 
 
 gr.Interface(
@@ -61,5 +61,6 @@ gr.Interface(
     title=title,
     description=description,
     flagging_options=["save"],
-    flagging_callback=hf_writer
+    flagging_callback=hf_writer,
+    allow_flagging=True,
 ).launch()