Update app.py
Browse files
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 |
-
|
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)
|