Update app.py

app.py

@@ -60,7 +60,7 @@ pipe.enable_model_cpu_offload()
 pipe.enable_xformers_memory_efficient_attention()
 
 # Generator seed,
-generator = torch.manual_seed(42)
+generator = torch.manual_seed(0)
 
 def get_canny_filter(image):
     if not isinstance(image, np.ndarray):
@@ -78,6 +78,7 @@ def generate_images(prompt, canny_image):
     output = pipe(
         prompt,
         canny_image,
+        seed=42000,
         generator=generator,
         num_images_per_prompt=1,
         num_inference_steps=20,
@@ -136,9 +137,9 @@ def infer():
 
     canny_image.save("canny1.jpg")
     canny_image2.save("canny2.jpg")
-    input_frame_1 = read_image(str("canny2.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_1 = read_image(str("canny1.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")
-    input_frame_2 = read_image(str("canny1.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_2 = read_image(str("canny2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
     
     #img1_batch = torch.stack([frames[0]])