Vitrous commited on
Commit
17247a8
·
verified ·
1 Parent(s): 7600265

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
  """