Update Dockerfile

Dockerfile

@@ -1,25 +1,29 @@
-# Use official Python base image
-FROM python:3.10-slim
+# Use official PyTorch image with CUDA support if GPU, or CPU-only
+ARG BASE_IMAGE=python:3.10-slim
+FROM ${BASE_IMAGE}
 
 # Set working directory
 WORKDIR /app
 
-# Copy requirements
-COPY requirements.txt .
-
-# Install dependencies
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
-    build-essential \
-    libsndfile1 \
+    git \
     ffmpeg \
+    libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Install Python packages
-RUN pip install --upgrade pip
+# Copy requirements
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy app code
 COPY . .
 
-# Set default command to run your app
+# Expose port for Gradio
+EXPOSE 7860
+
+# Start the app
 CMD ["python", "app.py"]