1

app.py

@@ -46,14 +46,15 @@ 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("https://huggingface.co/spaces/besarismaili/sdanim/tree/main/output")
+    output_dir = os.path.join(repo_dir, "output")
+    image_path = "/tmp/frames/"
 
     # Render each frame of animation
     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.Files(value=file_path)
+        yield gr.Image(value=file_path)
       
 with gr.Blocks() as demo:
     gr.Markdown("Stability Animation")
@@ -61,7 +62,7 @@ with gr.Blocks() as demo:
     s_promt = gr.Textbox(label="Second Prompt", value="a photo of a cute dog")
     stability_key = gr.Textbox(label="Stability Key", value="")
     
-    outimg = gr.File(label="Generated Files")
+    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")