Update app.py

app.py

@@ -131,9 +131,14 @@ def infer():
     
     diffused_img[1].save("diffused_input1.jpg")
     
-    input_frame_1 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
+    pil2diff_img2 = Image.open("./frame1.jpg")
+    canny_image2 = get_canny_filter(pil2diff_img2)
+
+    canny_image.save("canny1.jpg")
+    canny_image2.save("canny2.jpg")
+    input_frame_1 = read_image(str("canny2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")
-    input_frame_2 = read_image(str("diffused_input1.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_2 = read_image(str("canny1.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
     
     #img1_batch = torch.stack([frames[0]])