Update app.py

app.py

@@ -114,7 +114,7 @@ def main():
                 else:
                     gen = set_seed(seed)
 
-                    with trace(pipe, load_heads=True) as tc:
+                    with trace(pipe, load_heads=False) as tc:
                         out2 = pipe(new_prompt, num_inference_steps=inf_steps, generator=gen)
                         image2 = np.array(out2.images[0]) / 255
             else: