Update app.py
Browse files
app.py
CHANGED
@@ -128,7 +128,7 @@ def hermes_generate_response(msg_prompt: str) -> dict:
|
|
128 |
<|im_start|>assistant
|
129 |
'''
|
130 |
pipe_output = pipe(prompt_template)[0]['generated_text']
|
131 |
-
assistant_reply = pipe_output.split('\n\n')[1]
|
132 |
|
133 |
return {"user": msg_prompt, "assistant": pipe_output}
|
134 |
except Exception as e:
|
@@ -430,6 +430,7 @@ async def hermes_chat(request: Request):
|
|
430 |
raise
|
431 |
except Exception as e:
|
432 |
raise HTTPException(status_code=500, detail=str(e))
|
|
|
433 |
@app.get('/get_response/{thread_id}')
|
434 |
async def get_response(thread_id: int):
|
435 |
"""
|
|
|
128 |
<|im_start|>assistant
|
129 |
'''
|
130 |
pipe_output = pipe(prompt_template)[0]['generated_text']
|
131 |
+
#assistant_reply = pipe_output.split('\n\n')[1]
|
132 |
|
133 |
return {"user": msg_prompt, "assistant": pipe_output}
|
134 |
except Exception as e:
|
|
|
430 |
raise
|
431 |
except Exception as e:
|
432 |
raise HTTPException(status_code=500, detail=str(e))
|
433 |
+
|
434 |
@app.get('/get_response/{thread_id}')
|
435 |
async def get_response(thread_id: int):
|
436 |
"""
|