Spaces:
Sleeping
Sleeping
Update ml_service.py
Browse files- ml_service.py +6 -6
ml_service.py
CHANGED
|
@@ -6,17 +6,17 @@ import pandas as pd
|
|
| 6 |
import spacy
|
| 7 |
from simpletransformers.ner import NERModel
|
| 8 |
from spacy.matcher import PhraseMatcher
|
| 9 |
-
from
|
| 10 |
-
from django.conf import settings
|
| 11 |
from emoji import demojize
|
| 12 |
import unicodedata
|
| 13 |
|
| 14 |
|
| 15 |
-
base_directory = settings.BASE_DIR
|
| 16 |
|
| 17 |
-
labels_file = f"
|
| 18 |
-
ner_model_directory = f"
|
| 19 |
-
sentiment_model_file = f"
|
| 20 |
|
| 21 |
|
| 22 |
class MlProcessing:
|
|
|
|
| 6 |
import spacy
|
| 7 |
from simpletransformers.ner import NERModel
|
| 8 |
from spacy.matcher import PhraseMatcher
|
| 9 |
+
from constants import POSITIVE_SENTIMENT_PATTERNS, LABEL_COLOR, CATEGORY_THRESHOLD
|
| 10 |
+
# from django.conf import settings
|
| 11 |
from emoji import demojize
|
| 12 |
import unicodedata
|
| 13 |
|
| 14 |
|
| 15 |
+
# base_directory = settings.BASE_DIR
|
| 16 |
|
| 17 |
+
labels_file = f"ml_models/labels.json"
|
| 18 |
+
ner_model_directory = f"ml_models/ner_model/"
|
| 19 |
+
sentiment_model_file = f"ml_models/sentiment_model/model.ft"
|
| 20 |
|
| 21 |
|
| 22 |
class MlProcessing:
|