Update app.py

app.py

@@ -19,28 +19,19 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16
 
 step = 4
-base = "black-forest-labs/FLUX.1-dev"
+model_id = "black-forest-labs/FLUX.1-dev"
 
 adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2")
 
+pipe = AnimateDiffPipeline.from_pretrained(
+    model_id, motion_adapter=adapter
+    ,token=os.getenv("hf_token")
+)
 scheduler = DDIMScheduler.from_pretrained(
-    base,
-    subfolder="scheduler",
-    clip_sample=False,
-    timestep_spacing="linspace",
-    beta_schedule="linear",
-    steps_offset=1,
-    token=os.getenv("hf_token")
+    model_id, subfolder="scheduler", clip_sample=False, timestep_spacing="linspace", steps_offset=1
+    ,token=os.getenv("hf_token")
 )
-
-pipe = AnimateDiffPipeline.from_pretrained(
-    base,
-    motion_adapter=adapter,
-    scheduler=scheduler,
-    torch_dtype=dtype,
-    token=os.getenv("hf_token")
-).to(device)
-
+pipe.scheduler = scheduler
 pipe.enable_vae_slicing()
 pipe.enable_vae_tiling()