Update Dockerfile

Dockerfile

@@ -1,11 +1,36 @@
-# Use the base image from your Docker container
-FROM trcoot/cpu-casuallm:latest
+# Use the full Python 3.9 image (if you need specific modules)
+FROM python:3.9.19
+
+# Working Directory
+WORKDIR /app
+
+COPY models/ /app/models/
+COPY app.py .
+
+# RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir torch==2.2.2
+RUN pip install --no-cache-dir packaging
+
+# Copy Dependencies (if you have any)
+COPY requirements.txt .
+
+# Install Dependencies (if you have any)
+RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install -U git+https://github.com/sustcsonglin/flash-linear-attention
+
+# Copy Custom Modules (Adjust paths if needed)
+COPY causal-conv1d/ /app/causal-conv1d/
+RUN cd /app/causal-conv1d && python setup.py install
+
+COPY mamba/ /app/mamba/
+RUN cd /app/mamba && python setup.py install
 
 # Set the environment variable for the Hugging Face cache directory
-ENV HF_HOME=/app/.cache 
+ENV HF_HOME=/app/.cache
 
 # Create the cache directory and give the appropriate permissions
 RUN mkdir -p /app/.cache && chmod 777 /app/.cache
 
-# Start the application (modify this based on your app's entry point)
-CMD ["python", "app.py"] 
+# Print Messages
+# CMD ["bash"]
+CMD ["python", "app.py"]