Refactor : access to the model

Dockerfile

@@ -17,6 +17,9 @@ ENV HF_HOME /code/.cache/huggingface
 # Create cache directory with appropriate permissions
 RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
 
+# Copy the model files into the image
+COPY ./bertModel-20240428T144909Z-001 /code/bertModel-20240428T144909Z-001
+
 # Copy the rest of the application files
 COPY . .
 

helper_functions.py

@@ -6,15 +6,13 @@ from typing import Optional
 from torch import Tensor
 
 # Load the model
-model_path = "Sayado/Model_PFE"
-model = BertForSequenceClassification.from_pretrained(model_path)
+model = BertForSequenceClassification.from_pretrained(r"bertModel-20240428T144909Z-001\bertModel\saved_model")
 
 # Load the tokenizer
-tokenizer_path = "Sayado/Model_PFE" 
-tokenizer = BertTokenizer.from_pretrained(tokenizer_path)
+tokenizer = BertTokenizer.from_pretrained(r"bertModel-20240428T144909Z-001\bertModel\saved_tokenizer")
 
 # Charger le label encoder
-with open("label_encoder.pkl", "rb") as f:
+with open(r"bertModel-20240428T144909Z-001\bertModel\label_encoder.pkl", "rb") as f:
     label_encoder = pickle.load(f)
 
 class_labels = {