Update app.py

app.py

@@ -47,12 +47,9 @@ if torch.cuda.is_available():
         previewer = Previewer()
         previewer_state_dict = torch.load("previewer/previewer_v1_100k.pt", map_location=torch.device('cpu'))["state_dict"]
         previewer.load_state_dict(previewer_state_dict)
-        def callback_prior(i, t, latents, kwargs):
-            print("i", i)
-            print("t", t)
-            print("latents", latents)
-            print("kwargs", kwargs)
-            print(i, t, latents, kwargs)
+        def callback_prior(self_remote, i, t, kwargs):
+            latents = kwargs["latents"]
+            print(latents)
             output = previewer(latents)
             output = numpy_to_pil(output.clamp(0, 1).permute(0, 2, 3, 1).float().cpu().numpy())
             return output