Update app.py

app.py

@@ -66,12 +66,16 @@ def translate(text,lang):
     print(ret)
     return ret
 
+progress((0, step))
+def progress_callback(i, t, z):
+    progress((i+1, step))
+
 def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
 
 @spaces.GPU(duration=40)
-def Piper(name,posi,neg):
+def Piper(name,posi):
     global step
     
     print("starting piper")
@@ -81,7 +85,9 @@ def Piper(name,posi,neg):
         height=512,
         width=512,
         num_inference_steps=step,
-        guidance_scale=1
+        guidance_scale=1,
+        callback=progress_callback,
+        callback_step=1
     )
 
     export_to_gif(out.frames[0],name)
@@ -173,8 +179,8 @@ def main():
     repo = "ByteDance/SDXL-Lightning"
     ckpt = f"sdxl_lightning_{step}step_unet.safetensors"
 
-    unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
-    unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device="cuda"))
+    #unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, dtype)
+    #unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device))
 
     repo = "ByteDance/AnimateDiff-Lightning"
     ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
@@ -182,7 +188,7 @@ def main():
     adapter = MotionAdapter().to(device, dtype)
     adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
 
-    pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, unet=unet, torch_dtype=dtype, variant="fp16").to(device=device, dtype=dtype)
+    pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype, variant="fp16").to(device)
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
     
     mp.set_start_method("spawn", force=True)