Update app.py

app.py

@@ -9,7 +9,7 @@ import gradio as gr
 import numpy as np
 from lxml.html import fromstring
 from pathos.threading import ThreadPool as Pool
-from diffusers import MotionAdapter, EulerDiscreteScheduler
+from diffusers import AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler
 from diffusers.pipelines.flux import FluxPipeline
 from diffusers.utils import export_to_gif
 from huggingface_hub import hf_hub_download
@@ -26,7 +26,8 @@ base = "black-forest-labs/FLUX.1-dev"
 
 adapter = MotionAdapter().to(device, dtype)
 adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
-pipe = FluxPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
+pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
+pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
 
 def translate(text,lang):