Update text_generator.py

text_generator.py

@@ -16,7 +16,8 @@ class TextGenerationTool(Tool):
 
         # Initialize the text generation pipeline
         #text_generator = pipeline("text-generation") llama mistralai/Mistral-7B-Instruct-v0.1
-        text_generator = pipeline(model="meta-llama/Llama-2-7b-chat-hf")
+        text_generator = pipeline(model="gpt2")
+        #text_generator = pipeline(model="meta-llama/Llama-2-7b-chat-hf")
 
         # Generate text based on a prompt
         generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)