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, GPT2LMHeadModel,TextGenerationPipeline
|
| 11 |
-
tokenizer = BertTokenizer.from_pretrained("
|
| 12 |
-
model = GPT2LMHeadModel.from_pretrained("
|
| 13 |
text_generator = TextGenerationPipeline(model, tokenizer)
|
| 14 |
|
| 15 |
def poem(cls):
|
|
|
|
| 8 |
'''
|
| 9 |
import gradio as gr
|
| 10 |
from transformers import BertTokenizer, GPT2LMHeadModel,TextGenerationPipeline
|
| 11 |
+
tokenizer = BertTokenizer.from_pretrained("./")
|
| 12 |
+
model = GPT2LMHeadModel.from_pretrained("./")
|
| 13 |
text_generator = TextGenerationPipeline(model, tokenizer)
|
| 14 |
|
| 15 |
def poem(cls):
|