rafmacalaba commited on
Commit
2935cea
·
1 Parent(s): 9a57f89

use ai4data

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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/gliner-initial-model"
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)