Hugging Face's logo

Spaces:
ai4data
/
datause-detector
Running

App Files Community
rafmacalaba commited on
Commit
4f3e097
·
1 Parent(s): ffaa1f8

change 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-v3"
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-v6-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)