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 #!/usr/bin/env python3
""" """
GLiNER NER Service - Zero-shot entity extraction GLiNER NER Service - Zero-shot entity extraction
Port: 9094, GPU 5 Port: 9093, GPU 5
""" """
import os import os
import re import re
@ -25,7 +25,7 @@ app.add_middleware(
# Global model # Global model
model = None 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(): def load_model():
@ -174,4 +174,4 @@ def extract_relations(text: str, entities: List[Entity]) -> List[Relation]:
if __name__ == "__main__": if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=9094) uvicorn.run(app, host="0.0.0.0", port=9093)