requirements.txt

transformers
torch
gradio

app.py

@@ -0,0 +1,24 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load GPT-2 text generation pipeline
+generator = pipeline("text-generation", model="gpt2")
+
+# Function to generate text
+def generate_text(prompt, max_length=100):
+    result = generator(prompt, max_new_tokens=max_length, do_sample=True, temperature=0.7)
+    return result[0]["generated_text"]
+
+# Gradio interface
+iface = gr.Interface(
+    fn=generate_text,
+    inputs=[
+        gr.Textbox(lines=2, placeholder="Type your prompt here...", label="Prompt"),
+        gr.Slider(10, 500, value=100, step=10, label="Max Tokens")
+    ],
+    outputs=gr.Textbox(label="Generated Text"),
+    title="GPT-2 Text Generator",
+    description="Type a prompt and GPT-2 will continue your text!"
+)
+
+iface.launch()