Update app.py

app.py

@@ -18,19 +18,6 @@ bases = {
     "Anime": "Yntec/mistoonAnime2"
 }
 
-motion_models = {
-    "Default": None,
-    "Zoom in": "guoyww/animatediff-motion-lora-zoom-in",
-    "Zoom out": "guoyww/animatediff-motion-lora-zoom-out",
-    "Tilt up": "guoyww/animatediff-motion-lora-tilt-up",
-    "Tilt down": "guoyww/animatediff-motion-lora-tilt-down",
-    "Pan left": "guoyww/animatediff-motion-lora-pan-left",
-    "Pan right": "guoyww/animatediff-motion-lora-pan-right",
-    "Roll left": "guoyww/animatediff-motion-lora-rolling-anticlockwise",
-    "Roll right": "guoyww/animatediff-motion-lora-rolling-clockwise",
-}
-
-
 # Preload models
 if not torch.cuda.is_available():
     raise NotImplementedError("No GPU detected!")
@@ -38,31 +25,28 @@ if not torch.cuda.is_available():
 device = "cuda"
 dtype = torch.float16
 
+motion_loaded = None
+
 pipes = {}
 for base_name, base_path in bases.items():
     pipe = AnimateDiffPipeline.from_pretrained(base_path, torch_dtype=dtype).to(device)
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
     pipes[base_name] = pipe
 
-# Load motion models
-for motion_name, motion_path in motion_models.items():
-    if motion_path:
-        motion_model = MotionAdapter.from_pretrained(motion_path, torch_dtype=dtype).to(device)
-        motion_models[motion_name] = motion_model
-
-
 # Function 
 @spaces.GPU(duration=60,queue=False)
 def generate_image(prompt, base="Realistic", motion="Default", step=8, progress=gr.Progress()):
     global pipes
-    global motion_models
+    global motion_loaded
 
     pipe = pipes[base]
 
-    if motion != "Default":
-        pipe.motion_adapter = motion_models[motion]
-    else:
-        pipe.motion_adapter = None
+    if motion_loaded != motion:
+        pipe.unload_lora_weights()
+        if motion != "":
+            pipe.load_lora_weights(motion, adapter_name="motion")
+            pipe.set_adapters(["motion"], [0.7])
+        motion_loaded = motion
 
 
     # Load step model if not already loaded