updates

Dockerfile

@@ -20,6 +20,7 @@ RUN mkdir -p /app/.cache && \
     chmod -R 777 /app && \
     chmod -R 777 /app/.cache
 ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_HOME=/app/.cache
     # Copy the rest of the application code
 COPY app.py /app/app.py
 COPY webchat.py /app/webchat.py

webchat.py

@@ -80,15 +80,13 @@ def get_model_test(model_type, max_tokens, min_tokens, decoding, temperature):
     return model
 
 # Download the model first
-model = SentenceTransformer('all-MiniLM-L6-v2')
-
+model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 # Embedding function
 class MiniLML6V2EmbeddingFunction(EmbeddingFunction):
     MODEL = model
     def __call__(self, texts):
         return MiniLML6V2EmbeddingFunction.MODEL.encode(texts).tolist()
 
-
 def extract_text(url):
     try:
         # Send an HTTP GET request to the URL