Create app.py

app.py

@@ -0,0 +1,16 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load a very small text-generation model
+generator = pipeline("text-generation", model="distilgpt2")
+
+def generate(text):
+    result = generator(text, max_length=60, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Textbox(label="Write a sentence"),
+    outputs=gr.Textbox(label="AI output"),
+    title="Tiny Text Generator"
+).launch()