Update main.py

main.py

@@ -1,7 +1,7 @@
 import torch
 import os
 import uuid
-from diffusers import AnimateDiffPipeline, MotionAdapter, EulerDiscreteScheduler
+from diffusers import AnimateDiffPipeline, EulerDiscreteScheduler
 from diffusers.utils import export_to_video
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
@@ -18,6 +18,17 @@ bases = {
     "3d": "Lykon/DreamShaper",
     "Anime": "Yntec/mistoonAnime2"
 }
+motion_choices = {
+    "": "Default",
+    "guoyww/animatediff-motion-lora-zoom-in": "Zoom in",
+    "guoyww/animatediff-motion-lora-zoom-out": "Zoom out",
+    "guoyww/animatediff-motion-lora-tilt-up": "Tilt up",
+    "guoyww/animatediff-motion-lora-tilt-down": "Tilt down",
+    "guoyww/animatediff-motion-lora-pan-left": "Pan left",
+    "guoyww/animatediff-motion-lora-pan-right": "Pan right",
+    "guoyww/animatediff-motion-lora-rolling-anticlockwise": "Roll left",
+    "guoyww/animatediff-motion-lora-rolling-clockwise": "Roll right"
+}
 step_loaded = None
 base_loaded = "Realistic"
 motion_loaded = None
@@ -39,7 +50,7 @@ def generate_image(prompt, base="Realistic", motion="", step=8):
     global step_loaded
     global base_loaded
     global motion_loaded
-    print(prompt, base, step)
+    print(prompt, base, step, motion)
 
     if step_loaded != step:
         repo = "ByteDance/AnimateDiff-Lightning"
@@ -82,4 +93,4 @@ async def generate_video(
 # Run the app
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=8000)