Update app.py

app.py

@@ -86,11 +86,11 @@ def generate_video(prompt, seed, image=None, task_type=None):
             return "Error: Model not initialized.", "{}"
 
         output = predictor.inference(**kwargs)
-        #save_dir = f"./result/{task_type}"
-        #os.makedirs(save_dir, exist_ok=True)
-        #video_out_file = f"{save_dir}/{prompt[:100].replace('/','')}_{int(seed)}.mp4"
-        #print(f"Generating video, local path: {video_out_file}")
-        #export_to_video(output, video_out_file, fps=24)
+        save_dir = f"./result/{task_type}"
+        os.makedirs(save_dir, exist_ok=True)
+        video_out_file = f"{save_dir}/{prompt[:100].replace('/','')}_{int(seed)}.mp4"
+        print(f"Generating video, local path: {video_out_file}")
+        export_to_video(output, video_out_file, fps=24)
         return output, str(kwargs)
 
     except Exception as e: