Chris4K commited on
Commit
6329f26
1 Parent(s): 3c99288

Update text_generator.py

Browse files
Files changed (1) hide show
  1. text_generator.py +2 -1
text_generator.py CHANGED
@@ -15,7 +15,8 @@ class TextGenerationTool(Tool):
15
  #generated_text = f"Generated text based on the prompt: '{prompt}'"
16
 
17
  # Initialize the text generation pipeline
18
- text_generator = pipeline(model="bigcode/starcoder")
 
19
 
20
  # Generate text based on a prompt
21
  generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)
 
15
  #generated_text = f"Generated text based on the prompt: '{prompt}'"
16
 
17
  # Initialize the text generation pipeline
18
+ text_generator = pipeline("text-generation")
19
+ #text_generator = pipeline(model="bigcode/starcoder")
20
 
21
  # Generate text based on a prompt
22
  generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)