Update Dockerfile

Dockerfile

@@ -1,21 +1,34 @@
-FROM python:3.9-slim
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
-WORKDIR /app
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
 
+# Install Python and dependencies
 RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
+    python3 python3-pip ffmpeg curl git wget \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+RUN ln -s /usr/bin/python3 /usr/bin/python
+RUN pip install --upgrade pip
+
+# Create app directory
+WORKDIR /app
 
-RUN pip3 install -r requirements.txt
+# Copy requirements and install
+COPY requirements.txt .
+RUN pip install -r requirements.txt
 
-EXPOSE 8501
+# Copy the app code into the image
+COPY . .
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Ensure the custom interface is in place
+# Optional as already in src/, but ensures it's updated
+RUN wget -O src/custome_interface.py https://huggingface.co/Jzuluaga/accent-id-commonaccent_xlsr-en-english/resolve/main/custom_interface.py
+
+# Pulling LLaMA model (optional pre-pull)
+RUN curl -fsSL https://ollama.com/install.sh | sh && ollama pull llama3.1
+
+EXPOSE 8501
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+CMD ["streamlit", "run", "streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]