Upload app.py

app.py

@@ -7,10 +7,6 @@ def generate(text):
     result = generator(text, max_length=300, temperature=0.7, num_return_sequences=1, do_sample=True)
     return result[0]["generated_text"]
 
-demo = gr.Interface(
-    fn=generate,
-    inputs = gr.inputs.Textbox(lines=3, label="Input Text"),
-    outputs=gr.outputs.Textbox(label="Generated Text"),
-)
+demo = gr.ChatInterface(generate)
 
 demo.launch()