test gradio

app.py

@@ -2,13 +2,17 @@ import gradio as gr
 import torch
 from diffusers import AnimateDiffPipeline, DDIMScheduler, MotionAdapter
 from diffusers.utils import export_to_gif
-from PIL import Image
-import io
 
-# Load the motion adapter
-adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2", torch_dtype=torch.float16)
+# Set device to CPU
+device = torch.device("cpu")
+
+# Load the motion adapter on CPU
+adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2", torch_dtype=torch.float32).to(device)
 model_id = "SG161222/Realistic_Vision_V5.1_noVAE"
-pipe = AnimateDiffPipeline.from_pretrained(model_id, motion_adapter=adapter, torch_dtype=torch.float16)
+pipe = AnimateDiffPipeline.from_pretrained(
+    model_id, motion_adapter=adapter, torch_dtype=torch.float32
+).to(device)
+
 scheduler = DDIMScheduler.from_pretrained(
     model_id,
     subfolder="scheduler",
@@ -18,8 +22,8 @@ scheduler = DDIMScheduler.from_pretrained(
     steps_offset=1,
 )
 pipe.scheduler = scheduler
+
 pipe.enable_vae_slicing()
-pipe.enable_model_cpu_offload()
 
 # Define the animation function
 def generate_animation(prompt, negative_prompt, num_frames, guidance_scale, num_inference_steps):