Update app.py

app.py

@@ -85,15 +85,15 @@ def text2vid(input_text):
             generator=torch.Generator("cpu").manual_seed(0)
         )
         frames = output.frames[0]
-        export_to_gif(frames, f"data/{index+1}.gif")
+        export_to_gif(frames, f"./{index+1}.gif")
 
 # 定义生成最终视频的函数
 def video_generate():
     frame_rate = 24
-    gif_files = sorted(glob.glob('./data/*.gif'))
+    gif_files = sorted(glob.glob('./*.gif'))
     clips = [VideoFileClip(gif) for gif in gif_files]
     final_clip = concatenate_videoclips(clips, method="compose")
-    final_clip.write_videofile('data/output_video.mp4', codec='libx264')
+    final_clip.write_videofile('output_video.mp4', codec='libx264')
 
 # 定义文本到音频函数
 def text2audio(text_input, duration_seconds):
@@ -102,14 +102,14 @@ def text2audio(text_input, duration_seconds):
     inputs = processor(text=[text_input], padding=True, return_tensors="pt")
     max_new_tokens = int((duration_seconds / 5) * 256)
     audio_values = model.generate(**inputs, max_new_tokens=max_new_tokens)
-    scipy.io.wavfile.write("data/bgm.wav", rate=model.config.audio_encoder.sampling_rate, data=audio_values[0, 0].numpy())
+    scipy.io.wavfile.write("bgm.wav", rate=model.config.audio_encoder.sampling_rate, data=audio_values[0, 0].numpy())
 
 # 定义生成结果视频的函数
 def result_generate():
-    video = VideoFileClip("data/output_video.mp4")
-    audio = AudioFileClip("data/bgm.wav")
+    video = VideoFileClip("output_video.mp4")
+    audio = AudioFileClip("bgm.wav")
     video = video.set_audio(audio)
-    video.write_videofile("output/result.mp4", codec="libx264", audio_codec="aac")
+    video.write_videofile("result.mp4", codec="libx264", audio_codec="aac")
 
 # 定义删除所有文件的函数
 def delete_all_files(directory):
@@ -132,12 +132,12 @@ def generate_video(image):
     sentences = text2text(text)
     text2vid(sentences)
     video_generate()
-    video = VideoFileClip("data/output_video.mp4")
+    video = VideoFileClip("output_video.mp4")
     duration = video.duration
     audio_text = text2text(text)
     text2audio(audio_text, duration)
     result_generate()
-    return "output/result.mp4"
+    return "result.mp4"
 
 # 定义 Gradio 接口
 # interface = gr.Interface(