Create app.py

app.py

@@ -0,0 +1,71 @@
+import gradio as gr
+from transformers import pipeline
+
+generator = pipeline("text-generation", model="distilbert/distilgpt2", device=-1)
+
+def generate_text(prompt, temperature, max_tokens, repetition_penalty):
+    if not prompt.strip():
+        return "Please enter a prompt."
+    result = generator(
+        prompt,
+        max_new_tokens=int(max_tokens),
+        temperature=float(temperature),
+        repetition_penalty=float(repetition_penalty),
+        do_sample=True,
+        truncation=True,
+    )
+    return result[0]["generated_text"]
+
+demo = gr.Interface(
+    fn=generate_text,
+    inputs=[
+        gr.Textbox(
+            label="Prompt",
+            placeholder="Start typing about birds...",
+            lines=3,
+            value="The northern cardinal sings",
+        ),
+        gr.Slider(
+            minimum=0.1,
+            maximum=2.0,
+            value=1.0,
+            step=0.1,
+            label="Temperature",
+            info="Lower = more predictable, Higher = more creative",
+        ),
+        gr.Slider(
+            minimum=20,
+            maximum=200,
+            value=80,
+            step=10,
+            label="Max New Tokens",
+            info="How much text to generate",
+        ),
+        gr.Slider(
+            minimum=1.0,
+            maximum=2.0,
+            value=1.2,
+            step=0.1,
+            label="Repetition Penalty",
+            info="Higher = less repetition",
+        ),
+    ],
+    outputs=gr.Textbox(label="Generated Text", lines=8),
+    title="Bird Text Generator",
+    description=(
+        "A text generation playground using distilgpt2, customized with bird-themed prompts. "
+        "This is Riley's Space 1 — a baseline text generator. It can write about birds "
+        "but doesn't actually know anything about them. Try different temperature settings "
+        "to see how randomness affects the output."
+    ),
+    examples=[
+        ["The northern cardinal sings", 1.0, 80, 1.2],
+        ["In the forest canopy, the warbler", 0.7, 100, 1.2],
+        ["Migration patterns suggest that", 0.5, 120, 1.0],
+        ["The red-tailed hawk circled above the parking lot", 1.0, 80, 1.2],
+        ["At dawn, the chorus of birdsong", 1.3, 100, 1.2],
+    ],
+    theme=gr.themes.Soft(),
+)
+
+demo.launch()