update config to support gradio 5

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import torch
 import os
-import spaces
 import uuid
 
 from diffusers import AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler
@@ -11,11 +10,6 @@ from safetensors.torch import load_file
 from PIL import Image
 from transformers import CLIPFeatureExtractor
 
-# Custom message for linking to other demos
-MORE_INFO = """ ## Related Demos
-        ### Check out other interesting demos here
-        """
-
 # Model configuration
 BASES = {
     "Cartoon": "frankjoshua/toonyou_beta6",
@@ -53,7 +47,6 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(
 feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
 
 # Main generation function
-@spaces.GPU(duration=30, queue=False)
 def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Progress()):
     global step_loaded
     global base_loaded
@@ -90,17 +83,13 @@ def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Prog
         motion_loaded = motion
 
     # Progress tracking
-    progress((0, step))
-    def progress_callback(i, t, z):
-        progress((i+1, step))
+    progress(0, desc="Starting generation...")
 
     # Generate the video
     output = pipe(
         prompt=prompt,
         guidance_scale=1.2,
-        num_inference_steps=step,
-        callback=progress_callback,
-        callback_steps=1
+        num_inference_steps=step
     )
 
     # Save and return the video
@@ -110,81 +99,80 @@ def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Prog
     return path
 
 # Gradio Interface
-with gr.Blocks(css="style.css") as demo:
-    # Header
-    gr.HTML(
+with gr.Blocks(theme=gr.themes.Default()) as demo:
+    gr.Markdown(
         """
-        <h1><center>Instant⚡Video Generator</center></h1>
-        <p><center><span class='warning'>Adjust steps from 4 to 8 for better results if needed.</span></center></p>
-        <p><center><strong>First generation takes longer, subsequent generations are faster.</strong></p>
-        <p><center>Follow the example prompts format for best results</p>
+        # Instant⚡Video Generator
+        
+        > **Note**: Adjust steps from 4 to 8 for better results if needed.
+        
+        First generation takes longer, subsequent generations are faster.
+        Follow the example prompts format for best results.
         """
     )
 
-    # Input interface
-    with gr.Group():
-        with gr.Row():
-            prompt = gr.Textbox(label='Prompt')
-        
-        with gr.Row():
-            select_base = gr.Dropdown(
-                label='Base model',
-                choices=list(BASES.keys()),
-                value=base_loaded,
-                interactive=True
-            )
-            
-            select_motion = gr.Dropdown(
-                label='Motion',
-                choices=[
-                    ("Default", ""),
-                    ("Zoom in", "guoyww/animatediff-motion-lora-zoom-in"),
-                    ("Zoom out", "guoyww/animatediff-motion-lora-zoom-out"),
-                    ("Tilt up", "guoyww/animatediff-motion-lora-tilt-up"),
-                    ("Tilt down", "guoyww/animatediff-motion-lora-tilt-down"),
-                    ("Pan left", "guoyww/animatediff-motion-lora-pan-left"),
-                    ("Pan right", "guoyww/animatediff-motion-lora-pan-right"),
-                    ("Roll left", "guoyww/animatediff-motion-lora-rolling-anticlockwise"),
-                    ("Roll right", "guoyww/animatediff-motion-lora-rolling-clockwise"),
-                ],
-                value="guoyww/animatediff-motion-lora-zoom-in",
-                interactive=True
+    with gr.Row():
+        with gr.Column(scale=4):
+            prompt = gr.Textbox(
+                label="Enter your prompt",
+                placeholder="Focus: Describe what you want to animate..."
             )
             
-            select_step = gr.Dropdown(
-                label='Inference steps',
-                choices=[
-                    ('1-Step', 1), 
-                    ('2-Step', 2),
-                    ('4-Step', 4),
-                    ('8-Step', 8),
-                ],
-                value=4,
-                interactive=True
-            )
-            
-            submit = gr.Button(
-                scale=1,
-                variant='primary'
+            with gr.Row():
+                select_base = gr.Dropdown(
+                    label='Base model',
+                    choices=list(BASES.keys()),
+                    value=base_loaded
+                )
+                
+                select_motion = gr.Dropdown(
+                    label='Motion Style',
+                    choices=[
+                        ("Default", ""),
+                        ("Zoom in", "guoyww/animatediff-motion-lora-zoom-in"),
+                        ("Zoom out", "guoyww/animatediff-motion-lora-zoom-out"),
+                        ("Tilt up", "guoyww/animatediff-motion-lora-tilt-up"),
+                        ("Tilt down", "guoyww/animatediff-motion-lora-tilt-down"),
+                        ("Pan left", "guoyww/animatediff-motion-lora-pan-left"),
+                        ("Pan right", "guoyww/animatediff-motion-lora-pan-right"),
+                        ("Roll left", "guoyww/animatediff-motion-lora-rolling-anticlockwise"),
+                        ("Roll right", "guoyww/animatediff-motion-lora-rolling-clockwise"),
+                    ],
+                    value=""
+                )
+                
+                select_step = gr.Dropdown(
+                    label='Quality Steps',
+                    choices=[
+                        ('1-Step (Fastest)', 1), 
+                        ('2-Step (Fast)', 2),
+                        ('4-Step (Balanced)', 4),
+                        ('8-Step (Best)', 8),
+                    ],
+                    value=4
+                )
+
+            generate_btn = gr.Button("Generate Video", variant="primary")
+
+        with gr.Column(scale=6):
+            video_output = gr.Video(
+                label="Generated Animation",
+                height=512,
+                width=512,
+                autoplay=True
             )
 
-    # Output video display
-    video = gr.Video(
-        label='Generated Animation',
-        autoplay=True,
-        height=512,
-        width=512,
-        elem_id="video_output"
+    # Event handlers
+    generate_btn.click(
+        fn=generate_image,
+        inputs=[prompt, select_base, select_motion, select_step],
+        outputs=video_output
     )
 
-    # Event handlers
-    gr.on(
-        triggers=[submit.click, prompt.submit],
+    prompt.submit(
         fn=generate_image,
         inputs=[prompt, select_base, select_motion, select_step],
-        outputs=[video],
-        api_name="instant_video",
-        queue=False
+        outputs=video_output
     )
 
     # Example prompts
@@ -199,11 +187,11 @@ with gr.Blocks(css="style.css") as demo:
             ["Focus: Kids Playing (Season: Winter)"],
             ["Focus: Cars in Street (Season: Rain, Daytime) (Shot from Distance) (Movement: Cars running)"]
         ],
+        inputs=prompt,
+        outputs=video_output,
         fn=generate_image,
-        inputs=[prompt],
-        outputs=[video],
-        cache_examples="lazy",
+        cache_examples=True,
     )
 
 # Launch the interface
-demo.queue().launch()
+demo.launch()

requirements.txt

@@ -1,8 +1,10 @@
-gradio==4.16.0
-torch>=2.0.0
+--extra-index-url https://download.pytorch.org/whl/cu118
+gradio==5.16.2
+torch==2.0.1+cu118
+torchvision==0.15.2+cu118
 diffusers==0.25.0
 transformers==4.36.2
+accelerate==0.25.0
 safetensors==0.4.1
 huggingface_hub==0.20.3
-Pillow==10.2.0
-spaces==0.19.3
+Pillow==10.2.0