Spaces:
Running
Running
Commit
·
cce0575
1
Parent(s):
4c028c5
change model
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ from gliner import GLiNER
|
|
12 |
#from relation_extraction import CustomGLiNERRelationExtractor
|
13 |
|
14 |
# Cache and initialize model + relation extractor
|
15 |
-
DATA_MODEL_ID = "rafmacalaba/gliner_re_finetuned-
|
16 |
model = GLiNER.from_pretrained(DATA_MODEL_ID, cache_dir=_CACHE_DIR)
|
17 |
from relation_extraction import CustomGLiNERRelationExtractor
|
18 |
relation_extractor = CustomGLiNERRelationExtractor(model=model, return_index=True)
|
|
|
12 |
#from relation_extraction import CustomGLiNERRelationExtractor
|
13 |
|
14 |
# Cache and initialize model + relation extractor
|
15 |
+
DATA_MODEL_ID = "rafmacalaba/gliner_re_finetuned-v7-pos"
|
16 |
model = GLiNER.from_pretrained(DATA_MODEL_ID, cache_dir=_CACHE_DIR)
|
17 |
from relation_extraction import CustomGLiNERRelationExtractor
|
18 |
relation_extractor = CustomGLiNERRelationExtractor(model=model, return_index=True)
|