Spaces:

llmahmad
/

test_app_text_generation

Sleeping

llmahmad commited on Jul 22, 2024

Commit

c9d9455

verified ·

1 Parent(s): 7264b6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,5 @@
 # Load the GPT-2 model and tokenizer
 model_name = 'gpt2-large'
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)

 # Load the GPT-2 model and tokenizer
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 model_name = 'gpt2-large'
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)