Update app.py
Browse files
app.py
CHANGED
@@ -8,8 +8,8 @@ FilePath: /zuo/app.py
|
|
8 |
'''
|
9 |
import gradio as gr
|
10 |
from transformers import BertTokenizer, TFGPT2LMHeadModel,TextGenerationPipeline
|
11 |
-
tokenizer = BertTokenizer.from_pretrained("
|
12 |
-
model = TFGPT2LMHeadModel.from_pretrained("
|
13 |
text_generator = TextGenerationPipeline(model, tokenizer)
|
14 |
|
15 |
def poem(cls):
|
|
|
8 |
'''
|
9 |
import gradio as gr
|
10 |
from transformers import BertTokenizer, TFGPT2LMHeadModel,TextGenerationPipeline
|
11 |
+
tokenizer = BertTokenizer.from_pretrained(".")
|
12 |
+
model = TFGPT2LMHeadModel.from_pretrained(".")
|
13 |
text_generator = TextGenerationPipeline(model, tokenizer)
|
14 |
|
15 |
def poem(cls):
|