Update app.py

app.py

@@ -6,11 +6,16 @@ from huggingface_hub import HfFolder, login
 from io import StringIO
 
 # Load GPT-2 model and tokenizer
-tokenizer_gpt2 = GPT2Tokenizer.from_pretrained('gpt2')
+
+# Set environment variable to avoid floating-point errors
+os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
+
+# Load the tokenizer and model
+tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
 
 # Create a pipeline for text generation using GPT-2
-text_generator = pipeline("text-generation", model=model_gpt2)
+text_generator = pipeline("text-generation", model=model_gpt2, tokenizer=tokenizer)
 
 # Load the LLaMA tokenizer
 tokenizer_llama = LlamaTokenizer.from_pretrained("meta-llama/Meta-Llama-3.1-8B")