download-modesl

Dockerfile

@@ -1,11 +1,6 @@
 FROM slabstech/dhwani-server-base
 WORKDIR /app
 
-COPY dhwani_config.json .
-
-# Create a directory for pre-downloaded models
-
-
 RUN mkdir -p /app/models
 
 # Define build argument for HF_TOKEN
@@ -18,6 +13,7 @@ ENV HF_TOKEN=$HF_TOKEN_DOCKER
 COPY download_models.py .
 RUN python download_models.py
 
+COPY dhwani_config.json .
 COPY . .
 
 # Set up user

Dockerfile.app

@@ -2,6 +2,7 @@
 FROM slabstech/dhwani-model-server:latest
 WORKDIR /app
 
+COPY dhwani_config.json .
 # Copy application code
 COPY . .
 

Dockerfile.models

@@ -1,5 +1,5 @@
 # Base image with CUDA support
-FROM nvidia/cuda:12.8.0-cudnn-devel-ubuntu22.04 AS model-downloader
+FROM slabstech/dhwani-model-server:latest
 
 
 # Create a directory for pre-downloaded models

download_models.py

@@ -12,7 +12,7 @@ if not hf_token:
 models = {
     #'llm_model': ('google/gemma-3-4b-it', Gemma3ForConditionalGeneration, AutoProcessor),
     #'tts_model': ('ai4bharat/IndicF5', AutoModel, None),
-    'asr_model': ('ai4bharat/indic-conformer-600m-multilingual', AutoModel, None),
+    #'asr_model': ('ai4bharat/indic-conformer-600m-multilingual', AutoModel, None),
     'trans_en_indic': ('ai4bharat/indictrans2-en-indic-dist-200M', AutoModelForSeq2SeqLM, AutoTokenizer),
     'trans_indic_en': ('ai4bharat/indictrans2-indic-en-dist-200M', AutoModelForSeq2SeqLM, AutoTokenizer),
     'trans_indic_indic': ('ai4bharat/indictrans2-indic-indic-dist-320M', AutoModelForSeq2SeqLM, AutoTokenizer),