changes

Dockerfile

@@ -2,10 +2,16 @@ FROM python:3.10
 
 WORKDIR /code
 
+# Copy requirements and install dependencies
 COPY ./requirements.txt /code/requirements.txt
-
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+# Set up the cache directory for Hugging Face models
+ENV TRANSFORMERS_CACHE=/code/.cache
+RUN mkdir -p /code/.cache && chmod -R 777 /code/.cache
+
+# Copy the application files
 COPY . .
 
-CMD ["gunicorn", "-b", "0.0.0.0:7860","app"]
+# Start the Flask app with Gunicorn
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "app"]

app.py

@@ -9,6 +9,9 @@ from history import load_dataset, get_unique_next_words_from_dataset
 from dotenv import load_dotenv
 import os
 
+os.environ["TRANSFORMERS_CACHE"] = "/code/.cache"
+
+
 from typing import List, Dict, Optional, Union
 import logging
 from most_repeted_sentences import sentences_name, get_most_repeated_sentences, save_most_repeated_sentences