Skip to content

Commit fe454d6

Browse files
fix: Flair (#703)
1 parent 5c74411 commit fe454d6

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

label_studio_ml/examples/flair/docker-compose.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,6 @@ services:
1414
- WORKERS=1
1515
- THREADS=8
1616
- LOG_LEVEL=DEBUG
17-
- FLAIR_MODEL_NAME=ner
17+
- FLAIR_MODEL_NAME=ner-multi
1818
ports:
1919
- "9090:9090"

label_studio_ml/examples/flair/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
logger = logging.getLogger(__name__)
1111

12-
FLAIR_MODEL_NAME = os.getenv("FLAIR_MODEL_NAME", "ner")
12+
FLAIR_MODEL_NAME = os.getenv("FLAIR_MODEL_NAME", "ner-multi")
1313
logger.info(f"Loading Flair model {FLAIR_MODEL_NAME}")
1414
_model = Classifier.load(FLAIR_MODEL_NAME)
1515

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
scipy==1.10.1
2-
flair==0.13.1
2+
torch==2.5.1
3+
flair==0.15

0 commit comments

Comments
 (0)