Nepjune commited on
Commit
93e78fe
·
verified ·
1 Parent(s): f1b8b86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,9 +8,9 @@ blip_model = BlipForConditionalGeneration.from_pretrained(model_id)
8
  blip_processor = BlipProcessor.from_pretrained(model_id)
9
 
10
  # Initialize TTS model from Hugging Face
11
- model_name = "facebook/tts-crdnn-baker-softmax"
12
- tts_tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  tts_model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
14
  tts = pipeline(task="text2speech", model=tts_model, tokenizer=tts_tokenizer)
15
 
16
  def generate_caption(image):
 
8
  blip_processor = BlipProcessor.from_pretrained(model_id)
9
 
10
  # Initialize TTS model from Hugging Face
11
+ model_name = "tts-mozilla/tts-ljspeech-multilingual"
 
12
  tts_model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
13
+ tts_tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  tts = pipeline(task="text2speech", model=tts_model, tokenizer=tts_tokenizer)
15
 
16
  def generate_caption(image):