Update app.py
Browse files
app.py
CHANGED
@@ -29,6 +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 |
|
33 |
# Create a pipeline for text generation using GPT-2
|
34 |
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 |
+
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)
|