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
|