Eros483 commited on
Commit
c8d1019
·
unverified ·
1 Parent(s): bd5c316

Switch GLiNER to fastino/gliner2-base-v1 (PyTorch/GPU), drop onnxruntime

Browse files
Files changed (2) hide show
  1. app.py +2 -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: temsa/eurocivic-gliner2-onnx-cpu")
27
  from gliner import GLiNER
28
 
29
- _gliner_model = GLiNER.from_pretrained("temsa/eurocivic-gliner2-onnx-cpu")
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
- onnxruntime>=1.20.0
 
 
3
  sentence-transformers>=5.4.0
4
  gliner>=0.3.0
5
  pydantic>=2.13.0
6
+ torch>=2.4.0
7
+