fix: wrong infer function

app.py

@@ -215,7 +215,7 @@ TKG-DM🥚🍚: Training-free Chroma Key Content Generation Diffusion Model
 
     gr.on(
         triggers=[run_button.click, prompt.submit],
-        fn=infer,
+        fn=on_generate,
         inputs=[
             prompt,
             negative_prompt,

tkg.py

@@ -22,8 +22,6 @@ def get_mean_shifted_latents(
         latent_channel = shifted_latents[:, idx, :, :]
 
         positive_ratio = (latent_channel > 0).float().mean()
-        print("positive_ratio", positive_ratio)
-        print("shift", shift, "sign", sign)
         target_ratio = positive_ratio + shift * sign
 
         # gradually shift latent_channel