Update app.py

app.py

@@ -36,13 +36,10 @@ def generate_text(input_text):
 
     output = llm(input_text)
     print(output)
-    cleaned_output_text = output_text.replace(input_text, "")
-    return cleaned_output_text
+    return output
 
 
 text_generation_interface = gr.Interface(
     fn=generate_text,
-    inputs=[
-        gr.inputs.Textbox(label="Input Text"),
-    ],
-    outputs=gr.inputs.Textbox(label="Generated Text")).launch()
+    inputs="text", outputs="text")
+text_generation_interface.launch()