Tmeena commited on
Commit
72ce43f
1 Parent(s): f195f8d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -24,7 +24,7 @@ RUN mkdir -p /app/cache /app/temp && chmod -R 777 /app/cache /app/temp
24
  RUN python -c "from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer; \
25
  M2M100ForConditionalGeneration.from_pretrained('facebook/m2m100_418M', cache_dir='/app/cache'); \
26
  M2M100Tokenizer.from_pretrained('facebook/m2m100_418M', cache_dir='/app/cache')" && \
27
- python -c "import whisper; whisper.load_model('small', download_root='/app/cache')"
28
 
29
  # Expose the necessary port (same as in the Flask app)
30
  EXPOSE 7860
 
24
  RUN python -c "from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer; \
25
  M2M100ForConditionalGeneration.from_pretrained('facebook/m2m100_418M', cache_dir='/app/cache'); \
26
  M2M100Tokenizer.from_pretrained('facebook/m2m100_418M', cache_dir='/app/cache')" && \
27
+ python -c "import whisper; whisper.load_model('turbo', download_root='/app/cache')"
28
 
29
  # Expose the necessary port (same as in the Flask app)
30
  EXPOSE 7860