Xu Xuenan commited on
Commit
3b33703
·
1 Parent(s): f7325de

Fix pathlib bug

Browse files
Files changed (1) hide show
  1. mm_story_agent/__init__.py +8 -4
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, config["story_dir"] / "speech")
40
 
41
  def generate_sound(self, config, pages):
 
42
  sound_agent = AudioLDM2Agent(config["sound_generation"])
43
- sound_agent.call(pages, config["story_dir"] / "sound")
44
 
45
  def generate_music(self, config, pages):
 
46
  music_agent = MusicGenAgent(config["music_generation"])
47
- music_agent.call(pages, config["story_dir"] / "music")
48
 
49
  def generate_image(self, config, pages):
 
50
  image_agent = StoryDiffusionAgent(config["image_generation"])
51
- image_agent.call(pages, config["story_dir"] / "image")
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]}