Spaces:
Sleeping
Sleeping
Commit
·
8720a33
1
Parent(s):
13bd40f
Went to an older version of Numpy
Browse files- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -8,7 +8,6 @@ from transformers import pipeline, AutoProcessor, AutoModelForVision2Seq
|
|
| 8 |
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig
|
| 9 |
import time
|
| 10 |
import nltk
|
| 11 |
-
from melo.api import TTS
|
| 12 |
import io
|
| 13 |
|
| 14 |
from transformers import BlipProcessor, BlipForConditionalGeneration
|
|
@@ -93,6 +92,7 @@ def generate_roast(caption, llm_components):
|
|
| 93 |
return response
|
| 94 |
|
| 95 |
def initialize_tts_model():
|
|
|
|
| 96 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 97 |
tts_model = TTS(language='EN', device=device)
|
| 98 |
speaker_ids = tts_model.hps.data.spk2id
|
|
|
|
| 8 |
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig
|
| 9 |
import time
|
| 10 |
import nltk
|
|
|
|
| 11 |
import io
|
| 12 |
|
| 13 |
from transformers import BlipProcessor, BlipForConditionalGeneration
|
|
|
|
| 92 |
return response
|
| 93 |
|
| 94 |
def initialize_tts_model():
|
| 95 |
+
from melo.api import TTS
|
| 96 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 97 |
tts_model = TTS(language='EN', device=device)
|
| 98 |
speaker_ids = tts_model.hps.data.spk2id
|
requirements.txt
CHANGED
|
@@ -3,7 +3,7 @@ torch==2.1.0
|
|
| 3 |
torchvision
|
| 4 |
transformers
|
| 5 |
pillow
|
| 6 |
-
numpy
|
| 7 |
accelerate
|
| 8 |
git+https://github.com/huggingface/diffusers
|
| 9 |
opencv-python
|
|
|
|
| 3 |
torchvision
|
| 4 |
transformers
|
| 5 |
pillow
|
| 6 |
+
numpy<2
|
| 7 |
accelerate
|
| 8 |
git+https://github.com/huggingface/diffusers
|
| 9 |
opencv-python
|