WatNeru commited on
Commit
8bb5b53
·
1 Parent(s): 03fb695

log周りの修正

Browse files
Files changed (2) hide show
  1. app.py +7 -1
  2. package/rust_adapter.py +1 -0
app.py CHANGED
@@ -216,6 +216,12 @@ def build_word_tree(payload: WordTreeRequest) -> List[WordTreeResponse]:
216
  if __name__ == "__main__":
217
  import uvicorn
218
 
219
- uvicorn.run(app, host="0.0.0.0", port=7860)
 
 
 
 
 
 
220
 
221
 
 
216
  if __name__ == "__main__":
217
  import uvicorn
218
 
219
+ uvicorn.run(
220
+ app,
221
+ host="0.0.0.0",
222
+ port=7860,
223
+ log_level=os.getenv("UVICORN_LOG_LEVEL", "warning"),
224
+ access_log=os.getenv("UVICORN_ACCESS_LOG", "false").lower() == "true",
225
+ )
226
 
227
 
package/rust_adapter.py CHANGED
@@ -48,6 +48,7 @@ class RustAdapter:
48
  top_k=top_k,
49
  max_depth=max_depth,
50
  )
 
51
  return [
52
  {"text": p.get_full_word(), "probability": float(p.probability)}
53
  for p in pieces
 
48
  top_k=top_k,
49
  max_depth=max_depth,
50
  )
51
+ print(f"[RustAdapter] build_word_tree: {pieces}")
52
  return [
53
  {"text": p.get_full_word(), "probability": float(p.probability)}
54
  for p in pieces