deploy

Dockerfile

@@ -7,6 +7,7 @@ WORKDIR /app
 # Copy all files to the container
 COPY . /app
 
+
 # Set HF_HOME to use an accessible cache directory
 ENV HF_HOME=/app/.cache
 RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache

service/prediction_service.py

@@ -9,7 +9,7 @@ hf_token = os.getenv('HF_TOKEN')
 # Hugging Face Model ID and local model directory
 hf_model_id = 'Alibaba-NLP/gte-base-en-v1.5'
 model_dir = '/tmp/sentence_transformer'  # Use /tmp for write permissions
-clf_model_path = '/tmp/logistic_regression_model.pkl'
+clf_model_path = 'models/logistic_regression_model.pkl'
 
 # Create model directory if not exists
 os.makedirs(model_dir, exist_ok=True)