Update Dockerfile

Dockerfile

@@ -1,11 +1,16 @@
 FROM python:3.9
 
+# Set working dir
 WORKDIR /app
-COPY requirements.txt .
 
+# Copy files
+COPY . /app
+
+# Install dependencies
 RUN pip install --upgrade pip
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+RUN pip install unsloth==0.0.19 unsloth_zoo==0.0.6
 RUN pip install -r requirements.txt
 
-COPY . .
-
+# Expose app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]