Commit
Browse files- app.py +2 -2
- requirements.txt +2 -1
app.py
CHANGED
@@ -20,8 +20,8 @@ translation_model = MBartForConditionalGeneration.from_pretrained(ckpt)
|
|
20 |
tokenizer.src_lang = 'en_XX'
|
21 |
|
22 |
# Load Audio Model
|
23 |
-
audio_processor = AutoProcessor.from_pretrained("suno/bark")
|
24 |
-
audio_model = AutoModel.from_pretrained("suno/bark")
|
25 |
|
26 |
|
27 |
# Methods
|
|
|
20 |
tokenizer.src_lang = 'en_XX'
|
21 |
|
22 |
# Load Audio Model
|
23 |
+
audio_processor = AutoProcessor.from_pretrained("suno/bark-small")
|
24 |
+
audio_model = AutoModel.from_pretrained("suno/bark-small")
|
25 |
|
26 |
|
27 |
# Methods
|
requirements.txt
CHANGED
@@ -4,4 +4,5 @@ accelerate
|
|
4 |
streamlit~=1.30.0
|
5 |
pillow~=10.3.0
|
6 |
requests~=2.31.0
|
7 |
-
ipython~=8.20.0
|
|
|
|
4 |
streamlit~=1.30.0
|
5 |
pillow~=10.3.0
|
6 |
requests~=2.31.0
|
7 |
+
ipython~=8.20.0
|
8 |
+
cv2
|