Spaces:

ai4data
/

datause-detector

Running

rafmacalaba commited on 19 days ago

Commit

d1de289

1 Parent(s): c8aa1a4

fix imports

Files changed (1) hide show

relation_extraction.py CHANGED Viewed

@@ -1,6 +1,7 @@
 from gliner.multitask.base import GLiNERBasePipeline
 from typing import Optional, List, Union
 from datasets import load_dataset, Dataset
 class CustomGLiNERRelationExtractor(GLiNERBasePipeline):
     """

 from gliner.multitask.base import GLiNERBasePipeline
 from typing import Optional, List, Union
 from datasets import load_dataset, Dataset
+from gliner import GLiNER
 class CustomGLiNERRelationExtractor(GLiNERBasePipeline):
     """