zuozuo commited on
Commit
7cdfa4e
·
1 Parent(s): 2175baa

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, 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):