joey1101 commited on
Commit
bb6f5bb
·
verified ·
1 Parent(s): a1a5f7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def load_models():
10
  # Load the text generation model
11
  story_model = pipeline("text-generation", model="gpt2") # Load pre-trained text generation model
12
  # Load the text-to-speech model
13
- tts_model = pipeline("text-to-speech", model="facebook/tts_transformer-es-css10") # Load a TTS model
14
  return caption_model, story_model, tts_model # Return all three models
15
 
16
  # Function to generate story from caption
 
10
  # Load the text generation model
11
  story_model = pipeline("text-generation", model="gpt2") # Load pre-trained text generation model
12
  # Load the text-to-speech model
13
+ tts_model = pipeline("text-to-speech", model="suno/bark") # Load a TTS model
14
  return caption_model, story_model, tts_model # Return all three models
15
 
16
  # Function to generate story from caption