Spaces:
Build error
Build error
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
|