Samuelblue commited on
Commit
cfdc5cf
·
1 Parent(s): f2c1b45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,7 +3,9 @@
3
  import gradio as gr
4
 
5
  # Initialize tokenizer and model
6
- tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 
 
7
  model = GPT2LMHeadModel.from_pretrained('gpt2')
8
 
9
  # Create a function to generate text
 
3
  import gradio as gr
4
 
5
  # Initialize tokenizer and model
6
+ tokenizer =
7
+ import transformers
8
+ tokenizer = transformers.GPT2Tokenizer.from_pretrained('gpt2')
9
  model = GPT2LMHeadModel.from_pretrained('gpt2')
10
 
11
  # Create a function to generate text