Update app.py

app.py

@@ -10,10 +10,6 @@ from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 from PIL import Image
 
-MORE = """ ## TRY Other Demos
-        ### Instant Image: 4k images in 5 Second -> https://huggingface.co/spaces/KingNish/Instant-Image
-        """
-
 # Constants
 bases = {
     "Cartoon": "frankjoshua/toonyou_beta6",
@@ -25,23 +21,14 @@ step_loaded = None
 base_loaded = "Realistic"
 motion_loaded = None
 
-# Ensure model and scheduler are initialized in GPU-enabled function
-if not torch.cuda.is_available():
-    raise NotImplementedError("No GPU detected!")
-
-device = "cuda"
-dtype = torch.float16
+# Set device to CPU
+device = "cpu"
+dtype = torch.float32
 pipe = AnimateDiffPipeline.from_pretrained(bases[base_loaded], torch_dtype=dtype).to(device)
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
 
-# Safety checkers
-from transformers import CLIPFeatureExtractor
-
-feature_extractor = CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
-
-# Function 
-@spaces.GPU(duration=30,queue=False)
-def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Progress()):
+@spaces.CPU(duration=30, queue=False)
+def generate_image(prompt, base="Realistic", motion="", step=8, duration=5, progress=gr.Progress()):
     global step_loaded
     global base_loaded
     global motion_loaded
@@ -72,30 +59,19 @@ def generate_image(prompt, base="Realistic", motion="", step=8, progress=gr.Prog
 
     name = str(uuid.uuid4()).replace("-", "")
     path = f"/tmp/{name}.mp4"
-    export_to_video(output.frames[0], path, fps=10)
+    export_to_video(output.frames[0], path, fps=10, duration=duration)
     return path
 
-
 # Gradio Interface
-with gr.Blocks(css="style.css") as demo:
-    gr.HTML(
-        "<h1><center>Mochi 1</center></h1>"
-      
-    )
+with gr.Blocks() as demo:
+    gr.HTML("<h1><center>AnimateDiff on CPU</center></h1>")
     with gr.Group():
         with gr.Row():
-            prompt = gr.Textbox(
-                label='Prompt'
-            )
+            prompt = gr.Textbox(label='Prompt')
         with gr.Row():
             select_base = gr.Dropdown(
                 label='Base model',
-                choices=[
-                    "Cartoon", 
-                    "Realistic",
-                    "3d",
-                    "Anime",
-                ],
+                choices=["Cartoon", "Realistic", "3d", "Anime"],
                 value=base_loaded,
                 interactive=True
             )
@@ -112,60 +88,36 @@ with gr.Blocks(css="style.css") as demo:
                     ("Roll left", "guoyww/animatediff-motion-lora-rolling-anticlockwise"),
                     ("Roll right", "guoyww/animatediff-motion-lora-rolling-clockwise"),
                 ],
-                value="guoyww/animatediff-motion-lora-zoom-in",
+                value="",
                 interactive=True
             )
             select_step = gr.Dropdown(
                 label='Inference steps',
-                choices=[
-                    ('1-Step', 1), 
-                    ('2-Step', 2),
-                    ('4-Step', 4),
-                    ('8-Step', 8),
-                ],
+                choices=[('1-Step', 1), ('2-Step', 2), ('4-Step', 4), ('8-Step', 8)],
                 value=4,
                 interactive=True
             )
-            submit = gr.Button(
-                scale=1,
-                variant='primary'
+            slider_duration = gr.Slider(
+                label='Video Duration (seconds)',
+                minimum=1,
+                maximum=10,
+                value=5,
+                step=1,
+                interactive=True
             )
+            submit = gr.Button(scale=1, variant='primary')
     video = gr.Video(
-        label='AnimateDiff-Lightning',
+        label='Generated Video',
         autoplay=True,
         height=512,
         width=512,
         elem_id="video_output"
     )
 
-    gr.on(triggers=[
-            submit.click,
-            prompt.submit
-    ],
-        fn = generate_image,
-        inputs = [prompt, select_base, select_motion, select_step],
-        outputs = [video],
-        api_name = "instant_video",
-        queue = False
-    )
-
-    gr.Examples(
-        examples=[
-        ["Focus: Eiffel Tower (Animate: Clouds moving)"], #Atmosphere Movement Example
-        ["Focus: Trees In forest (Animate: Lion running)"], #Object Movement Example
-        ["Focus: Astronaut in Space"], #Normal
-        ["Focus: Group of Birds in sky (Animate:  Birds Moving) (Shot From distance)"], #Camera distance
-        ["Focus:  Statue of liberty (Shot from Drone) (Animate: Drone coming toward statue)"], #Camera Movement
-        ["Focus: Panda in Forest (Animate: Drinking Tea)"], #Doing Something
-        ["Focus: Kids Playing (Season: Winter)"], #Atmosphere or Season
-        {"Focus: Cars in Street (Season: Rain, Daytime) (Shot from Distance) (Movement: Cars running)"} #Mixture
-    ], 
+    submit.click(
         fn=generate_image,
-        inputs=[prompt],
-        outputs=[video],
-        cache_examples="lazy",
-)
+        inputs=[prompt, select_base, select_motion, select_step, slider_duration],
+        outputs=[video]
+    )
 
 demo.queue().launch()
-
-Translate