diff --git a/indiquo/cli/IndiQuoCLI.py b/indiquo/cli/IndiQuoCLI.py index a9c0614f63c1c14d8202668e99aed5abaeafa491..0dba212ddbec90a02d9f4c4f16662cab5cff3e87 100644 --- a/indiquo/cli/IndiQuoCLI.py +++ b/indiquo/cli/IndiQuoCLI.py @@ -5,13 +5,14 @@ from datetime import datetime from os import listdir from os.path import join, isfile, splitext, basename, isdir from pathlib import Path -from flair.models import SequenceTagger +try: + from flair.models import SequenceTagger + from indiquo.core.CandidatePredictorRW import CandidatePredictorRW +except ModuleNotFoundError: + pass from quid.core.Quid import Quid - from dramatist.core.Dramatist import Dramatist - -from indiquo.core.CandidatePredictorRW import CandidatePredictorRW from indiquo.core.CandidatePredictorST import CandidatePredictorST from indiquo.core.IndiQuo import IndiQuo from indiquo.core.ScenePredictor import ScenePredictor @@ -19,9 +20,7 @@ from indiquo.core.chunker.SentenceChunker import SentenceChunker from indiquo.core.CandidatePredictor import CandidatePredictor from sentence_transformers import SentenceTransformer from transformers import AutoModelForSequenceClassification, AutoTokenizer - import csv - from indiquo.training.candidate import TrainCandidateClassifier diff --git a/indiquo/core/CandidatePredictorRW.py b/indiquo/core/CandidatePredictorRW.py index 65c8da456c7c08b082f076a6ccdedce92680a8c4..531f371e6b9f459311ede1d99da16a96aadffe24 100644 --- a/indiquo/core/CandidatePredictorRW.py +++ b/indiquo/core/CandidatePredictorRW.py @@ -1,7 +1,6 @@ from typing import List, Optional from flair.data import Sentence from flair.nn import Model - from indiquo.core.BasePredictor import BasePredictor from indiquo.core.Candidate import Candidate from indiquo.core.chunker.BaseChunker import BaseChunker