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 AnimateDiffSDXLPipeline, MotionAdapter, DDIMScheduler
+from diffusers import DiffusionPipeline, AnimateDiffSDXLPipeline, MotionAdapter, DDIMScheduler
 from diffusers.pipelines.flux import FluxPipeline
 from diffusers.utils import export_to_gif
 from huggingface_hub import hf_hub_download
@@ -24,6 +24,8 @@ ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 #base = "emilianJR/epiCRealism"
 base = "black-forest-labs/FLUX.1-dev"
 
+vid_pipe = DiffusionPipeline.from_pretrained(repo)
+
 adapter = MotionAdapter().to(device)
 adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt)))
 
@@ -34,16 +36,10 @@ scheduler = DDIMScheduler.from_pretrained(
     timestep_spacing="linspace",
     beta_schedule="linear",
     steps_offset=1,
-    token=os.getenv("hf_token")
 )
-flx = FluxPipeline.from_pretrained(
-    base,
-    torch_dtype=dtype,
-    token=os.getenv("hf_token")
-).to(device)
-pipe = AnimateDiffSDXLPipeline.from_pretrained(
+
+pipe = AnimateDiffPipeline(vid_pipe).from_pretrained(
     base,
-    unet=flx.components.unet,
     motion_adapter=adapter,
     scheduler=scheduler,
     torch_dtype=dtype,
@@ -53,6 +49,14 @@ pipe = AnimateDiffSDXLPipeline.from_pretrained(
 pipe.enable_vae_slicing()
 pipe.enable_vae_tiling()
 
+ """
+ flx = FluxPipeline.from_pretrained(
+    base,
+    torch_dtype=dtype,
+    token=os.getenv("hf_token")
+).to(device)
+ """
+
 def translate(text,lang):
     
     if text == None or lang == None: