Handle avi & ogg output formats

app.py

@@ -35,7 +35,7 @@ def animate(
     seed: Optional[int] = 42,
     randomize_seed: bool = True,
     motion_bucket_id: int = 127,
-    fps_id: int = 6,
+    fps_id: int = 25,
     noise_aug_strength: float = 0.1,
     decoding_t: int = 3,
     video_format: str = "mp4",
@@ -92,7 +92,22 @@ def animate(
     ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + \
     str(secondes) + " sec."
     
-    return gr.update(value=video_path, format=video_format if video_format != "gif" else None, visible=video_format != "gif"), gr.update(value=gif_path, visible=video_format == "gif"), gr.update(value=result_path, visible=True), gr.update(label="Generated frames in *." + frame_format + " format", format = frame_format, value = frames, visible=True), seed, gr.update(value = information, visible = True), gr.update(visible=True)
+    return [
+        # Display for video
+        gr.update(value = video_path, format = video_format if video_format != "gif" else None, visible=video_format != "gif"),
+        # Display for gif
+        gr.update(value = gif_path, visible = video_format == "gif"),
+        # Download button
+        gr.update(label = "💾 Download animation in *." + video_format + " format", value=result_path, visible=True),
+        # Frames
+        gr.update(label = "Generated frames in *." + frame_format + " format", format = frame_format, value = frames, visible = True),
+        # Used seed
+        seed,
+        # Information
+        gr.update(value = information, visible = True),
+        # Reset button
+        gr.update(visible = True)
+    ]
 
 @spaces.GPU(duration=120)
 def animate_on_gpu(
@@ -165,7 +180,7 @@ def reset():
 with gr.Blocks() as demo:
   gr.HTML("""
     <h1><center>Image-to-Video</center></h1>
-    <big><center>Animate your images into 25 frames of 1024x576 pixels freely, without account, without watermark and download the video</center></big>
+    <big><center>Animate your image into 25 frames of 1024x576 pixels freely, without account, without watermark and download the video</center></big>
     <br/>
     
     <p>
@@ -183,7 +198,7 @@ with gr.Blocks() as demo:
               motion_bucket_id = gr.Slider(label="Motion bucket id", info="Controls how much motion to add/remove from the image", value=127, minimum=1, maximum=255)
               noise_aug_strength = gr.Slider(label="Noise strength", info="The noise to add", value=0.1, minimum=0, maximum=1, step=0.1)
               decoding_t = gr.Slider(label="Decoding", info="Number of frames decoded at a time; this eats more VRAM; reduce if necessary", value=3, minimum=1, maximum=5, step=1)
-              video_format = gr.Radio([["*.mp4", "mp4"], ["*.gif", "gif"]], label="Video format for result", info="File extention", value="mp4", interactive=True)
+              video_format = gr.Radio([["*.mp4", "mp4"], ["*.avi", "avi"], ["*.ogg", "ogg"], ["*.gif", "gif"]], label="Video format for result", info="File extention", value="mp4", interactive=True)
               frame_format = gr.Radio([["*.webp", "webp"], ["*.png", "png"], ["*.jpeg", "jpeg"], ["*.gif (unanimated)", "gif"], ["*.bmp", "bmp"]], label="Image format for frames", info="File extention", value="webp", interactive=True)
               version = gr.Radio([["Auto", "auto"], ["🏃🏻‍♀️ SVD (trained on 14 f/s)", "svd"], ["🏃🏻‍♀️💨 SVD-XT (trained on 25 f/s)", "svdxt"], ["DragNUWA", "dragnuwa"]], label="Model", info="Trained model", value="auto", interactive=True)
               seed = gr.Slider(label="Seed", value=42, randomize=True, minimum=0, maximum=max_64_bit_int, step=1)
@@ -193,7 +208,7 @@ with gr.Blocks() as demo:
           reset_btn = gr.Button(value="🧹 Reinit page", variant="stop", elem_id="reset_button", visible = False)
 
       with gr.Column():
-          video_output = gr.Video(label="Generated video", autoplay=True)
+          video_output = gr.Video(label="Generated video", format="mp4", autoplay=True)
           gif_output = gr.Image(label="Generated video", format="gif", visible=False)
           download_button = gr.DownloadButton(label="💾 Download video", visible=False)
           information_msg = gr.HTML(visible=False)