deploy file added

.github/workflows/deploy.yml

@@ -0,0 +1,26 @@
+name: Deploy to Sagemaker
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+    deploy:
+        runs-on: ubuntu-latest
+
+        steps:
+            - name: Checkout code
+              uses: actions/checkout@v2
+
+            - name: Setup Python
+              uses: actions/setup-python@v2
+              with:
+                python-version: 3.10
+
+            - name: Login to AWS
+              uses: aws-actions/configure-aws-credentials@v2
+              with:
+                aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }}
+                aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
+                aws-region: ${{ secrets.AWS_REGION }}

README.md

@@ -1,3 +1,4 @@
 # Real Time Customer Support Chatbot
-Developing a real-time customer support chatbot using a fine-tuned LLM to provide accurate responses. Building CI/CD pipelines for scalable deployment on AWS SageMaker and integrating MLflow for tracking model versions, experiment logging, and continuous improvements.
+---
+### Developing a real-time customer support chatbot using a fine-tuned LLM to provide accurate responses. Building CI/CD pipelines for scalable deployment on AWS SageMaker and integrating MLflow for tracking model versions, experiment logging, and continuous improvements.
 ---

dockerfile

@@ -0,0 +1,30 @@
+# Use Python 3.10 slim as base image
+FROM python:3.10-slim
+
+# Set the working directory
+WORKDIR /app
+
+# Copy the current directory contents into the container at /app
+COPY . /app
+
+# Install dependencies
+RUN apt-get update && apt-get install -y \
+    python3-pip \
+    git
+
+# Install pip packages without caching
+RUN pip install --no-cache-dir --upgrade pip
+RUN pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy .env file to the working directory
+COPY .env /app/.env
+
+# Set environment variables from .env file
+ENV $(cat /app/.env | xargs)
+
+# Expose port 7860
+EXPOSE 7860
+
+# Run the application
+CMD ["python", "app.py"]

requirements.txt

@@ -1,4 +1,4 @@
-transformers=4.37
+transformers==4.37
 torch
 mlflow
 boto3

src/config.py

@@ -56,4 +56,5 @@ class Config:
     
     # DVC configurations
     DVC_REMOTE_NAME = "s3-storage"
-    DVC_REMOTE_URL = f"s3://{S3_BUCKET}/dvc"
+    DVC_REMOTE_URL = f"s3://{S3_BUCKET}/dvc"
+