personality detection models added to files.deeppavlov

deeppavlov · Dec 29, 2023 · 9c2d20c · 9c2d20c
1 parent db14239
commit 9c2d20c
Show file tree

Hide file tree

Showing 11 changed files with 14 additions and 11 deletions.
diff --git a/annotators/emotion_detection/Dockerfile b/annotators/emotion_detection/Dockerfile
@@ -36,6 +36,4 @@ RUN gdown 'https://drive.google.com/uc?id=15rB8cn3MSJpFNIKHY3f9qvCMbgH2ZLmR' -O
 RUN gdown 'https://drive.google.com/uc?id=1ZHw2MEbjLwU5A7EaNysgVHZlaWq4FJZG' -O '/data/bert-large-uncased_none_seed-42.pt'
 RUN gdown 'https://drive.google.com/uc?id=1XZVPI2ZvE0PbvhFMinsAp2z9SPPEsSjn' -O '/data/final_model.pt'
 RUN gdown 'https://drive.google.com/uc?id=1DMUHqbwqn-j9XIhoqnfSwqyqrlPMkVJe' -O '/data/XCLIP_Augmented.pt'
-RUN gdown 'https://drive.google.com/uc?id=1VxEQcso_Bsrzix1TykriJ2BLfUm9Gtx5' -O '/data/redundant_features.txt'
-
-RUN echo $REDUNDANT_FEATURES
+RUN gdown 'https://drive.google.com/uc?id=1VxEQcso_Bsrzix1TykriJ2BLfUm9Gtx5' -O '/data/redundant_features.txt'
diff --git a/annotators/personality_detection/Dockerfile b/annotators/personality_detection/Dockerfile
@@ -2,8 +2,14 @@ FROM python:3.7
 
 WORKDIR /src
 
+COPY . /src
+RUN mkdir /data
+
 COPY requirements.txt .
 
 RUN pip install -r requirements.txt
 
+RUN wget -O models http://files.deeppavlov.ai/dream_data/personality_detection/personality_detection_models_v0.tar.gz && tar -xf models
+RUN mv -f models_v0 /data/models && ls /data/models && rm -rf models_v0
+
 COPY . .
diff --git a/annotators/personality_detection/models/cAGR.p b/annotators/personality_detection/models/cAGR.p
diff --git a/annotators/personality_detection/models/cCON.p b/annotators/personality_detection/models/cCON.p
diff --git a/annotators/personality_detection/models/cEXT.p b/annotators/personality_detection/models/cEXT.p
diff --git a/annotators/personality_detection/models/cNEU.p b/annotators/personality_detection/models/cNEU.p
diff --git a/annotators/personality_detection/models/cOPN.p b/annotators/personality_detection/models/cOPN.p
diff --git a/annotators/personality_detection/models/vectorizer_30.p b/annotators/personality_detection/models/vectorizer_30.p
diff --git a/annotators/personality_detection/models/vectorizer_31.p b/annotators/personality_detection/models/vectorizer_31.p
diff --git a/annotators/personality_detection/server.py b/annotators/personality_detection/server.py
@@ -12,13 +12,13 @@
 
 sentry_sdk.init(os.getenv("SENTRY_DSN"))
 
-cEXT = pickle.load(open("./models/cEXT.p", "rb"))
-cNEU = pickle.load(open("./models/cNEU.p", "rb"))
-cAGR = pickle.load(open("./models/cAGR.p", "rb"))
-cCON = pickle.load(open("./models/cCON.p", "rb"))
-cOPN = pickle.load(open("./models/cOPN.p", "rb"))
-vectorizer_31 = pickle.load(open("./models/vectorizer_31.p", "rb"))
-vectorizer_30 = pickle.load(open("./models/vectorizer_30.p", "rb"))
+cEXT = pickle.load(open("/data/models/cEXT.p", "rb"))
+cNEU = pickle.load(open("/data/models/cNEU.p", "rb"))
+cAGR = pickle.load(open("/data/models/cAGR.p", "rb"))
+cCON = pickle.load(open("/data/models/cCON.p", "rb"))
+cOPN = pickle.load(open("/data/models/cOPN.p", "rb"))
+vectorizer_31 = pickle.load(open("/data/models/vectorizer_31.p", "rb"))
+vectorizer_30 = pickle.load(open("/data/models/vectorizer_30.p", "rb"))
 
 
 logger = logging.getLogger(__name__)

diff --git a/state_formatters/dp_formatters.py b/state_formatters/dp_formatters.py
@@ -511,7 +511,6 @@ def simple_formatter_service(payload: List):
     sent_rewrite_formatter, sent_segm_formatter, base_skill_selector_formatter
     """
     logging.info(f"answer {payload}")
-    logging.warning(f"answer {payload}")
     return payload