Update app.py

app.py

@@ -12,7 +12,7 @@ generator = pipeline(
 )
 model = AutoModelWithLMHead.from_pretrained("gpt2")
 def green(prompt):
-    text_generator(prompt, max_length=500)
+    return text_generator(prompt, max_length=500)
     
-iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs="text")
+iface = gr.Interface(fn=green, inputs=gr.inputs.Textbox(lines=2, placeholder="Put prompt here"), outputs=gr.outputs.Textbox(type="str", label="generated text"))
 iface.launch()