AnkitPatil commited on
Commit
c140288
Β·
1 Parent(s): 02d66b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -4,8 +4,13 @@ from transformers import pipeline
4
  generator = pipeline("text-generation", model = "distilgpt2")
5
 
6
 
7
- prompt = "This tutorial is about"
8
- generator(prompt, max_length = 30)
 
9
 
10
- iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
11
- iface.launch()
 
 
 
 
 
4
  generator = pipeline("text-generation", model = "distilgpt2")
5
 
6
 
7
+ def generate_text(prompt):
8
+ generated_text = generator(prompt, max_length=30, num_return_sequences=1)[0]['generated_text']
9
+ return generated_text
10
 
11
+ iface = gr.Interface(fn=generate_text, inputs="text",
12
+ outputs="text",
13
+ live=True,
14
+ title="Text Generator App πŸ“πŸ“πŸ“πŸ“",
15
+ description="Enter a text.")
16
+ iface.launch(share=True)