Upload folder using huggingface_hub

Dockerfile

@@ -4,11 +4,11 @@ WORKDIR /
 
 RUN pip install --no-cache-dir --upgrade -r /requirements.txt
 
-ENV KAFKA_URL=${KAFKA_URL}
-ENV KAFKA_API_KEY=${KAFKA_API_KEY}
-ENV KAFKA_IP=${KAFKA_IP}
+ENV KAFKA_URL=$KAFKA_URL
+ENV KAFKA_API_KEY=$KAFKA_API_KEY
+ENV KAFKA_IP=$KAFKA_IP
 
-RUN echo "KAFKA_URL=${KAFKA_URL}"
+RUN echo "KAFKA_URL=$KAFKA_URL"
 
 ENV TRANSFORMERS_CACHE=/transformers_cache
 RUN mkdir -p  /transformers_cache && chmod -R 777  /transformers_cache

classificator.py

@@ -1,6 +1,5 @@
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
-import numpy as np
 import pandas as pd
 import pickle 
 st = SentenceTransformer('all-mpnet-base-v2')