yakine commited on
Commit
1743f62
·
verified ·
1 Parent(s): 49b1c06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ if not hf_token:
29
  # Load GPT-2 model and tokenizer
30
  tokenizer_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
31
  model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
32
- model_gpt2.config.pad_token_id = model_gpt2.config.eos_token_id
33
 
34
  # Create a pipeline for text generation using GPT-2
35
  text_generator = pipeline("text-generation", model=model_gpt2, tokenizer=tokenizer_gpt2)
 
29
  # Load GPT-2 model and tokenizer
30
  tokenizer_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
31
  model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
32
+
33
 
34
  # Create a pipeline for text generation using GPT-2
35
  text_generator = pipeline("text-generation", model=model_gpt2, tokenizer=tokenizer_gpt2)