Update app.py

app.py

@@ -123,7 +123,9 @@ def infer():
    
     #frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
     #print(f"FRAME BEFORE stack: {frames[100]}")
+    
     prompt = "astronaut crew inside a spaceship, artwork by Claude Monet, beautiful details"
+    
     pil2diff_img = Image.open("./frame1.jpg")
     canny_image = get_canny_filter(pil2diff_img)
     diffused_img = generate_images(prompt, canny_image)
@@ -136,9 +138,9 @@ def infer():
 
     canny_image.save("canny1.jpg")
     canny_image2.save("canny2.jpg")
-    input_frame_1 = read_image(str("canny1.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_1 = read_image(str("diffused_input1.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")
-    input_frame_2 = read_image(str("canny2.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
     
     #img1_batch = torch.stack([frames[0]])
@@ -271,12 +273,12 @@ def infer():
     blend2 = Image.blend(res,blend2,0.5)
     blend2.save("blended2.jpg")
 
-    pil2diff_img = Image.open("blended2.jpg")
-    canny_image = get_canny_filter(pil2diff_img)
-    diffused_img = generate_images(prompt, canny_image)
-    print(f"DIFFUSED IMG: {diffused_img[1]}")
+    pil2diff_blend = Image.open("blended2.jpg")
+    canny_image = get_canny_filter(pil2diff_blend)
+    diffused_blend = generate_images(prompt, canny_image)
+    print(f"DIFFUSED IMG: {diffused_blend[1]}")
     
-    diffused_img[1].save("diffused_blended_2.jpg")
+    diffused_blend[1].save("diffused_blended_2.jpg")
     
     return "done", "predicted_flow.jpg", ["flofile.flo"], "diffused_input1.jpg", "diffused_blended_2.jpg", 'wraped.jpg', "blended2.jpg"
 ####################################