Spaces:
Paused
Paused
Commit
·
2935cea
1
Parent(s):
9a57f89
use ai4data
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ from gliner import GLiNER
|
|
13 |
|
14 |
# Cache and initialize model + relation extractor
|
15 |
# DATA_MODEL_ID = "rafmacalaba/gliner_re_finetuned-vfin"
|
16 |
-
DATA_MODEL_ID = "rafmacalaba/
|
17 |
model = GLiNER.from_pretrained(DATA_MODEL_ID, cache_dir=_CACHE_DIR)
|
18 |
from relation_extraction import CustomGLiNERRelationExtractor
|
19 |
relation_extractor = CustomGLiNERRelationExtractor(model=model, return_index=True)
|
|
|
13 |
|
14 |
# Cache and initialize model + relation extractor
|
15 |
# DATA_MODEL_ID = "rafmacalaba/gliner_re_finetuned-vfin"
|
16 |
+
DATA_MODEL_ID = "rafmacalaba/ai4data-datause-extraction"
|
17 |
model = GLiNER.from_pretrained(DATA_MODEL_ID, cache_dir=_CACHE_DIR)
|
18 |
from relation_extraction import CustomGLiNERRelationExtractor
|
19 |
relation_extractor = CustomGLiNERRelationExtractor(model=model, return_index=True)
|