Vitrous commited on
Commit
4b87e64
·
verified ·
1 Parent(s): 8752a28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -118,6 +118,7 @@ def hermes_generate_response(msg_prompt: str) -> dict:
118
  Returns:
119
  dict: A dictionary containing the user's message prompt and the model's response.
120
  """
 
121
  try:
122
  prompt_template=f'''<|im_start|>system
123
  {system_message}<|im_end|>
 
118
  Returns:
119
  dict: A dictionary containing the user's message prompt and the model's response.
120
  """
121
+ pipe = pipeline("text-generation",model=model, tokenizer=tokenizer, **generation_params)
122
  try:
123
  prompt_template=f'''<|im_start|>system
124
  {system_message}<|im_end|>