AIEONE commited on
Commit
2d7cd09
·
verified ·
1 Parent(s): bc9b630

Delete gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +0 -19
gradio_app.py DELETED
@@ -1,19 +0,0 @@
1
- import gradio as gr
2
- from transformers import pipeline
3
-
4
- model = pipeline("text-generation", model="distilgpt2")
5
-
6
- def generate_text(prompt):
7
- generated_text = model(prompt, max_length=50, num_return_sequences=1)
8
- return generated_text[0]['generated_text']
9
-
10
- iface = gr.Interface(
11
- fn=generate_text,
12
- inputs=gr.Textbox(lines=2, placeholder="Enter text here..."),
13
- outputs="text",
14
- title="Text Generation Model",
15
- description="Enter a prompt to generate text."
16
- )
17
-
18
- if __name__ == "__main__":
19
- iface.launch()