yumoqing/ner-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: model path to /data/ymq/models/gliner-multitask-large-v0.5

Browse Source
This commit is contained in:
yumoqing 2026-07-22 11:00:25 +08:00
parent e5317a1e82
commit 2d587dff1d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app.py
View File

@ -1,7 +1,7 @@
#!/usr/bin/env python3
"""
GLiNER NER Service - Zero-shot entity extraction
Port: 9094, GPU 5
Port: 9093, GPU 5
"""
import os
import re
@ -25,7 +25,7 @@ app.add_middleware(
# Global model
model = None
MODEL_PATH = "/mnt/disk0/yumoqing/models/gliner-multitask-large-v0.5"
MODEL_PATH = "/data/ymq/models/gliner-multitask-large-v0.5"
def load_model():
@ -174,4 +174,4 @@ def extract_relations(text: str, entities: List[Entity]) -> List[Relation]:
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=9094)
uvicorn.run(app, host="0.0.0.0", port=9093)