Spaces:
Running on Zero
Running on Zero
Eros483 commited on
Switch GLiNER to fastino/gliner2-base-v1 (PyTorch/GPU), drop onnxruntime
Browse files- app.py +2 -2
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -23,10 +23,10 @@ def _load_gliner():
|
|
| 23 |
global _gliner_model
|
| 24 |
if _gliner_model is not None:
|
| 25 |
return
|
| 26 |
-
logger.info("Loading GLiNER:
|
| 27 |
from gliner import GLiNER
|
| 28 |
|
| 29 |
-
_gliner_model = GLiNER.from_pretrained("
|
| 30 |
logger.info("GLiNER loaded")
|
| 31 |
|
| 32 |
|
|
|
|
| 23 |
global _gliner_model
|
| 24 |
if _gliner_model is not None:
|
| 25 |
return
|
| 26 |
+
logger.info("Loading GLiNER: fastino/gliner2-base-v1")
|
| 27 |
from gliner import GLiNER
|
| 28 |
|
| 29 |
+
_gliner_model = GLiNER.from_pretrained("fastino/gliner2-base-v1")
|
| 30 |
logger.info("GLiNER loaded")
|
| 31 |
|
| 32 |
|
requirements.txt
CHANGED
|
@@ -3,4 +3,5 @@ uvicorn>=0.46.0
|
|
| 3 |
sentence-transformers>=5.4.0
|
| 4 |
gliner>=0.3.0
|
| 5 |
pydantic>=2.13.0
|
| 6 |
-
|
|
|
|
|
|
| 3 |
sentence-transformers>=5.4.0
|
| 4 |
gliner>=0.3.0
|
| 5 |
pydantic>=2.13.0
|
| 6 |
+
torch>=2.4.0
|
| 7 |
+
|