docker file modified

Dockerfile

@@ -1,10 +1,10 @@
 FROM python:3.9
 
-ENV PATH="/home/user/.local/bin:$PATH"
-
 WORKDIR /code
-
 COPY . /code
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
+
+
 
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt

src/api/main.py

@@ -15,10 +15,6 @@ def home():
 def display_prediction(message : str = "Hello World"):
     try:
         dimention = logistic_regression.get_label(message)
-        # dimention = message
         return {"message" : dimention}
     except Exception as e:
-        return f"Unable to fetch the data {e}"
-
-# x = display_prediction()
-# print(type(x))
+        return f"Unable to fetch the data {e}"

src/modules/logistic_regression.py

@@ -10,7 +10,7 @@ src_directory = os.path.abspath(os.path.join(os.path.dirname(__file__), "../..",
 sys.path.append(src_directory)
 from data import sample_data
 
-file_path = r"src\data\sms_process_data_main.xlsx"
+file_path = r"src/data/sms_process_data_main.xlsx"
 df = sample_data.get_data_frame(file_path)
 
 def get_label(message):