log周りの修正
Browse files- app.py +7 -1
- 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(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|