1

app.py

@@ -46,9 +46,11 @@ def anim(f_promt, s_promt, stability_key):
     repo_dir = os.getenv('HF_COMET_ROOT_PATH', '/')
 
     # Define output folder path
-    output_dir = os.path.join(repo_dir, "output")
     image_path = "/tmp/frames/"
+    output_dir = os.path.join(image_path, "output")
 
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
     # Render each frame of animation
     for idx, frame in enumerate(animator.render()):
         file_path = os.path.join(output_dir, f"frame_{idx:05d}.png")