Update

app.py

@@ -30,8 +30,9 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, times
 # Function 
 @spaces.GPU(enable_queue=True)
 def generate_image(prompt, base, step):
-    global loaded
-    print(prompt, step)
+    global step_loaded
+    global base_loaded
+    print(prompt, base, step)
 
     if step_loaded != step:
         repo = "ByteDance/AnimateDiff-Lightning"