zuozuo commited on
Commit
7d94a8b
1 Parent(s): 43dd320

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("./tokenizer_config.json")
12
- model = TFGPT2LMHeadModel.from_pretrained("./config.json")
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):