Yuchan5386 commited on
Commit
9e90de0
·
verified ·
1 Parent(s): 7d6a990

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -205,4 +205,4 @@ async def generate_text_mirostat_top_p_with_buffer(model, prompt, max_len=100, m
205
  @app.get("/generate")
206
  async def generate(request: Request):
207
  prompt = request.query_params.get("prompt", "안녕하세요")
208
- return StreamingResponse(generate_text_mirostat_top_p_with_buffer(prompt), media_type="text/plain")
 
205
  @app.get("/generate")
206
  async def generate(request: Request):
207
  prompt = request.query_params.get("prompt", "안녕하세요")
208
+ return StreamingResponse(generate_text_mirostat_top_p_with_buffer(model, prompt), media_type="text/plain")