Jofthomas HF staff commited on
Commit
8996eaa
·
1 Parent(s): 0899d0f

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