Spaces:
Running
Running
Update app.py
Browse files
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="
|
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
|