wrote the code for text generation

app.py

@@ -0,0 +1,18 @@
+from transformers import pipeline
+import gradio as gr
+
+generator = pipeline("text-generation", model="gpt2")
+
+def generate_prompt(topic):
+    prompt = f"Write a creative story about {topic}. Here is a prompt to get you started:"
+    result = generator(prompt, max_length=50, num_return_sequences=1)
+    return result[0]['generated_text']
+
+iface = gr.Interface(
+    fn=generate_prompt,
+    input="text",
+    output="text",
+    title="Creative Writing Prompt Generation",
+)
+
+iface.launch()