Update app.py
Browse files
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|>
|