Spaces:
Running
on
L4
Running
on
L4
Xu Xuenan
commited on
Commit
·
3b33703
1
Parent(s):
f7325de
Fix pathlib bug
Browse files
mm_story_agent/__init__.py
CHANGED
@@ -35,20 +35,24 @@ class MMStoryAgent:
|
|
35 |
return pages
|
36 |
|
37 |
def generate_speech(self, config, pages):
|
|
|
38 |
speech_agent = CosyVoiceAgent(config["speech_generation"])
|
39 |
-
speech_agent.call(pages,
|
40 |
|
41 |
def generate_sound(self, config, pages):
|
|
|
42 |
sound_agent = AudioLDM2Agent(config["sound_generation"])
|
43 |
-
sound_agent.call(pages,
|
44 |
|
45 |
def generate_music(self, config, pages):
|
|
|
46 |
music_agent = MusicGenAgent(config["music_generation"])
|
47 |
-
music_agent.call(pages,
|
48 |
|
49 |
def generate_image(self, config, pages):
|
|
|
50 |
image_agent = StoryDiffusionAgent(config["image_generation"])
|
51 |
-
image_agent.call(pages,
|
52 |
|
53 |
def generate_modality_assets(self, config, pages):
|
54 |
script_data = {"pages": [{"story": page} for page in pages]}
|
|
|
35 |
return pages
|
36 |
|
37 |
def generate_speech(self, config, pages):
|
38 |
+
story_dir = Path(config["story_dir"])
|
39 |
speech_agent = CosyVoiceAgent(config["speech_generation"])
|
40 |
+
speech_agent.call(pages, story_dir / "speech")
|
41 |
|
42 |
def generate_sound(self, config, pages):
|
43 |
+
story_dir = Path(config["story_dir"])
|
44 |
sound_agent = AudioLDM2Agent(config["sound_generation"])
|
45 |
+
sound_agent.call(pages, story_dir / "sound")
|
46 |
|
47 |
def generate_music(self, config, pages):
|
48 |
+
story_dir = Path(config["story_dir"])
|
49 |
music_agent = MusicGenAgent(config["music_generation"])
|
50 |
+
music_agent.call(pages, story_dir / "music")
|
51 |
|
52 |
def generate_image(self, config, pages):
|
53 |
+
story_dir = Path(config["story_dir"])
|
54 |
image_agent = StoryDiffusionAgent(config["image_generation"])
|
55 |
+
image_agent.call(pages, story_dir / "image")
|
56 |
|
57 |
def generate_modality_assets(self, config, pages):
|
58 |
script_data = {"pages": [{"story": page} for page in pages]}
|