Jofthomas HF staff commited on
Commit
6774f29
·
verified ·
1 Parent(s): b2b6546

Update TextGen/router.py

Browse files
Files changed (1) hide show
  1. TextGen/router.py +1 -1
TextGen/router.py CHANGED
@@ -44,7 +44,7 @@ class Generate(BaseModel):
44
 
45
  def generate_text(messages: List[str]):
46
  print(messages)
47
- promptmessages[-1]
48
  prompt = PromptTemplate(template=prompt, input_variables=['Prompt'])
49
 
50
  # Initialize the LLM
 
44
 
45
  def generate_text(messages: List[str]):
46
  print(messages)
47
+ prompt=messages[-1]
48
  prompt = PromptTemplate(template=prompt, input_variables=['Prompt'])
49
 
50
  # Initialize the LLM