black44 commited on
Commit
5b88937
·
verified ·
1 Parent(s): b8e2812

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +0 -19
Dockerfile CHANGED
@@ -37,25 +37,6 @@ RUN pip install soundfile
37
  # Set user to non-root for Hugging Face compatibility
38
  USER user
39
 
40
- # Download Bark TTS model
41
- RUN python3 - <<EOF
42
- from transformers import AutoTokenizer, AutoProcessor, BarkModel
43
- model_name = "suno/bark-small"
44
- print(f"Downloading {model_name}...")
45
- AutoTokenizer.from_pretrained(model_name).save_pretrained("/app/models/suno-bark")
46
- AutoProcessor.from_pretrained(model_name).save_pretrained("/app/models/suno-bark")
47
- BarkModel.from_pretrained(model_name).save_pretrained("/app/models/suno-bark")
48
- EOF
49
-
50
- # Download sentiment analysis model
51
- RUN python3 - <<EOF
52
- from transformers import AutoModelForSequenceClassification, AutoTokenizer
53
- model_name = "cardiffnlp/twitter-xlm-roberta-base-sentiment"
54
- print(f"Downloading {model_name}...")
55
- AutoTokenizer.from_pretrained(model_name).save_pretrained("/app/models/sentiment")
56
- AutoModelForSequenceClassification.from_pretrained(model_name).save_pretrained("/app/models/sentiment")
57
- EOF
58
-
59
  # Copy application source code
60
  COPY --chown=user:user app.py .
61
 
 
37
  # Set user to non-root for Hugging Face compatibility
38
  USER user
39
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
  # Copy application source code
41
  COPY --chown=user:user app.py .
42