Update text_generator.py

text_generator.py

@@ -36,7 +36,8 @@ class TextGenerationTool(Tool):
         #generated_text = f"Generated text based on the prompt: '{prompt}'"
 
         # Initialize the text generation pipeline
-        text_generator = pipeline(model="lgaalves/gpt2-dolly", token=token)
+        #text_generator = pipeline(model="lgaalves/gpt2-dolly", token=token)
+        text_generator = pipeline(model="microsoft/Orca-2-13b", token=token)
 
         # Generate text based on a prompt
         generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)