Update Dockerfile

Dockerfile

@@ -1,37 +1,7 @@
-# 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 .
+# Use the base image from your Docker container
+FROM trcoot/cpu-casuallm:latest
 
-# 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
-
-# Create the cache directory and give the appropriate permissions
-RUN mkdir -p /app/.cache && chmod 777 /app/.cache
-RUN mkdir -p /app/gradio_flagged && chmod 777 /app/gradio_flagged
-
-# Print Messages
-# CMD ["bash"]
-CMD ["python", "app.py"]
+# Start the application (modify this based on your app's entry point)
+CMD ["python", "app.py"]