Hugging Face's logo

Spaces:
ai4data
/
datause-detector
Running

App Files Community
rafmacalaba commited on
Commit
201ab7f
·
1 Parent(s): cce0575

new model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-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)
 
12
  #from relation_extraction import CustomGLiNERRelationExtractor
13
 
14
  # Cache and initialize model + relation extractor
15
+ DATA_MODEL_ID = "rafmacalaba/gliner_re_finetuned-vfin"
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)