AlonG11 commited on
Commit
adc7ff3
1 Parent(s): dd96b87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import GPT2Tokenizer, TFGPT2LMHeadModel
6
  model = TFGPT2LMHeadModel.from_pretrained("fine-tuned-gpt2")
7
 
8
  # Initialize the tokenizer
9
- tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
10
 
11
  # Set the maximum length for the generated text
12
  max_length = 100
 
6
  model = TFGPT2LMHeadModel.from_pretrained("fine-tuned-gpt2")
7
 
8
  # Initialize the tokenizer
9
+ tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
10
 
11
  # Set the maximum length for the generated text
12
  max_length = 100