Kuberwastaken commited on
Commit
8720a33
·
1 Parent(s): 13bd40f

Went to an older version of Numpy

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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