Test

Dockerfile

@@ -19,7 +19,6 @@ ENV NKTL_DATA=/nltk_data
 RUN [ "python", "-c", "import nltk; nltk.download('stopwords', download_dir='/usr/local/nltk_data')" ]
 
 RUN chown -R 1000:1000 /code
-RUN chown -R 1000:1000 /.cache
 
 
 USER 1000

app.py

@@ -2,6 +2,7 @@ import pandas as pd
 from sentence_transformers import SentenceTransformer, util
 from flask import Flask, render_template, request, jsonify
 from nltk.corpus import stopwords
+import os
 stop = stopwords.words('english')
 
 
@@ -34,9 +35,9 @@ def df_to_text(df):
 df = pd.read_csv("data/dataset.csv").reset_index(drop=True)
 embedding_df = pd.read_csv("data/embedding.csv", header=None)
 docs = embedding_df.values
-# text = "a white shirt for men"
+HF_TOKEN=os.environ("HF_TOKEN")
 
-model = SentenceTransformer("bert-base-nli-mean-tokens")
+model = SentenceTransformer("bert-base-nli-mean-tokens", cache_folder = "/code/", use_auth_token=HF_TOKEN)
 
 
 app = Flask(__name__)