Update app.py
Browse files
app.py
CHANGED
@@ -4,8 +4,14 @@ from transformers import BertTokenizer, GPT2LMHeadModel,TextGenerationPipeline
|
|
4 |
tokenizer = BertTokenizer.from_pretrained("uer/gpt2-chinese-poem")
|
5 |
model = GPT2LMHeadModel.from_pretrained("uer/gpt2-chinese-poem")
|
6 |
text_generator = TextGenerationPipeline(model, tokenizer)
|
7 |
-
def greet(name):
|
8 |
-
return "Hello " + name + "!!"
|
9 |
|
10 |
-
|
11 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
tokenizer = BertTokenizer.from_pretrained("uer/gpt2-chinese-poem")
|
5 |
model = GPT2LMHeadModel.from_pretrained("uer/gpt2-chinese-poem")
|
6 |
text_generator = TextGenerationPipeline(model, tokenizer)
|
|
|
|
|
7 |
|
8 |
+
def predict(prompt):
|
9 |
+
completion = text_generator(prompt, max_length=50, do_sample=True)
|
10 |
+
return completion
|
11 |
+
|
12 |
+
gr.Interface(fn=predict, inputs="text", outputs="text").launch()
|
13 |
+
#def greet(name):
|
14 |
+
# return "Hello " + name + "!!"
|
15 |
+
|
16 |
+
#iface = gr.Interface(fn=greet, inputs="text", outputs="text")
|
17 |
+
#iface.launch()
|