| |
| """app.ipynb |
| |
| Automatically generated by Colab. |
| |
| Original file is located at |
| https://colab.research.google.com/drive/1-t9RYzLIxj4UDdky3Q_wyg29YwrPfJZN |
| """ |
|
|
| import gradio as gr |
| from transformers import pipeline |
|
|
| generator = pipeline("text-generation", model="gpt2") |
|
|
| def generate_text(prompt): |
| output = generator(prompt, max_new_tokens=50, do_sample=True, return_full_text=False) |
| return output[0]["generated_text"] |
|
|
| demo = gr.Interface( |
| fn=generate_text, |
| inputs=gr.Textbox(lines=4, placeholder="Type a short prompt here..."), |
| outputs=gr.Textbox(lines=4, label="Output"), |
| allow_flagging="never", |
| title="Text Generator Demo", |
| description="A simple Gradio app using GPT-2 from Hugging Face Transformers.", |
| theme="soft", |
| ) |
|
|
| demo.queue().launch() |
