Update Dockerfile

Dockerfile

@@ -1,6 +1,5 @@
 FROM python:3.11-slim
 
-# System dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     ffmpeg \
     git \
@@ -8,50 +7,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 WORKDIR /app
 
-# Install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
 COPY . .
 
-# Pre-download all models at build time using snapshot_download
-# HF_TOKEN must be set as a build secret or Space secret
-ARG HF_TOKEN
-ENV HF_TOKEN=${HF_TOKEN}
-
-RUN python - <<'EOF'
-import os
-from huggingface_hub import snapshot_download, login
-
-token = os.environ.get("HF_TOKEN", "")
-if token:
-    login(token=token)
-
-models = [
-    "drrobot9/wav2vec2-nigerian-language-identifier-v8",
-    "NCAIR1/Yoruba-ASR",
-    "NCAIR1/Igbo-ASR",
-    "NCAIR1/Hausa-ASR",
-    "NCAIR1/NigerianAccentedEnglish",
-    "NCAIR1/N-ATLaS",
-]
-
-for model_id in models:
-    print(f"Downloading {model_id}...")
-    snapshot_download(
-        repo_id=model_id,
-        token=token,
-        ignore_patterns=["*.msgpack", "*.h5", "flax_model*", "tf_model*",
-                         "rust_model*", "optimizer.pt", "rng_state*",
-                         "training_args.bin", "trainer_state.json",
-                         "scheduler.pt"],
-    )
-    print(f"  Done: {model_id}")
-
-print("All models downloaded.")
-EOF
-
 EXPOSE 7860
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["python", "start.py"]