aamirtaymoor commited on
Commit
7570912
·
verified ·
1 Parent(s): 5246eca

Update ml_service.py

Browse files
Files changed (1) hide show
  1. 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 einstein.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"{base_directory}/ml_models/labels.json"
18
- ner_model_directory = f"{base_directory}/ml_models/ner_model/"
19
- sentiment_model_file = f"{base_directory}/ml_models/sentiment_model/model.ft"
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: