Create app.py

app.py

@@ -0,0 +1,103 @@
+import gradio as gr
+from transformers import pipeline
+
+TITLE = "News Draft Generator"
+DESCRIPTION = (
+    "Experiment with news-style writing using distilgpt2. "
+    "Try headlines, article openings, research summaries, and briefings."
+)
+
+generator = pipeline("text-generation", model="distilgpt2")
+
+
+def generate_text(prompt: str, temperature: float, top_p: float, max_length: int) -> str:
+    if not prompt or not prompt.strip():
+        return "Please enter a prompt."
+
+    # Clamp temperature to avoid division-by-zero-style issues in downstream sampling
+    temperature = max(0.01, float(temperature))
+
+    outputs = generator(
+        prompt,
+        do_sample=True,
+        temperature=temperature,
+        top_p=float(top_p),
+        max_length=int(max_length),
+        num_return_sequences=1,
+        pad_token_id=generator.tokenizer.eos_token_id,
+    )
+
+    return outputs[0]["generated_text"]
+
+
+examples = [
+    ["Breaking news: scientists discovered that", 0.5, 0.9, 100],
+    ["A new report released today found that students", 0.5, 0.9, 100],
+    ["The conference on artificial intelligence announced", 0.5, 0.9, 100],
+    ["Researchers at the university published a study showing", 0.5, 0.9, 100],
+    ["This week in technology: the biggest story is", 0.5, 0.9, 100],
+]
+
+with gr.Blocks(title=TITLE) as demo:
+    gr.Markdown(f"# {TITLE}")
+    gr.Markdown(DESCRIPTION)
+
+    with gr.Row():
+        with gr.Column():
+            prompt = gr.Textbox(
+                label="Prompt",
+                placeholder="Enter a headline, opening sentence, or briefing prompt...",
+                lines=4,
+            )
+
+            temperature = gr.Slider(
+                minimum=0.1,
+                maximum=2.0,
+                value=0.5,
+                step=0.1,
+                label="Temperature",
+                info="Controls how creative/wild the writing is",
+            )
+
+            top_p = gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=0.9,
+                step=0.05,
+                label="Top-p",
+                info="Controls word diversity",
+            )
+
+            max_length = gr.Slider(
+                minimum=20,
+                maximum=200,
+                value=100,
+                step=1,
+                label="Max Length",
+                info="Controls how much text it generates",
+            )
+
+            generate_button = gr.Button("Generate")
+
+        with gr.Column():
+            output = gr.Textbox(
+                label="Generated Text",
+                lines=16,
+            )
+
+    generate_button.click(
+        fn=generate_text,
+        inputs=[prompt, temperature, top_p, max_length],
+        outputs=output,
+    )
+
+    gr.Examples(
+        examples=examples,
+        inputs=[prompt, temperature, top_p, max_length],
+        outputs=output,
+        fn=generate_text,
+        cache_examples=False,
+    )
+
+if __name__ == "__main__":
+    demo.launch()