1

app.py

@@ -52,11 +52,14 @@ def anim(f_promt, s_promt, stability_key):
     if not os.path.exists(output_dir):
         os.makedirs(output_dir)
     # Render each frame of animation
+    images = []
     for idx, frame in enumerate(animator.render()):
         file_path = os.path.join(output_dir, f"frame_{idx:05d}.png") 
         frame.save(file_path)
         print("Created frame at:"+file_path)
-        yield gr.Image(value=file_path)
+        images.append(gr.Image(value=file_path))
+    
+    return images
       
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")
@@ -67,6 +70,6 @@ with gr.Blocks() as demo:
     outimg = gr.Image(label="Generated Files")
     btn = gr.Button('Anim')
 
-    btn.click(fn=anim, inputs=[f_promt, s_promt, stability_key], outputs=[outimg],api_name="AnimAPI")
+    btn.click(fn=anim, inputs=[f_promt, s_promt, stability_key], outputs=[outimg],api_name="AnimAPI", result_format="list")
 
 demo.launch()