Update app.py

app.py

@@ -264,8 +264,14 @@ def infer():
     blend2 = Image.open('raw_frame2.jpg')
     blend2 = Image.blend(res,blend2,0.5)
     blend2.save("blended2.jpg")
+
+    pil2diff_img = Image.open("blended2.jpg")
+    diffused_img = generate_images(pil2diff_img, prompt)
+    print(f"DIFFUSED IMG: {diffused_img[1]}")
+    
+    diffused_img[1].save("diffused_blended_2.jpg")
     
-    return "done", "predicted_flow.jpg", ["flofile.flo"], "diffused_input1.jpg", 'raw_frame2.jpg', 'wraped.jpg', "blended2.jpg"
+    return "done", "predicted_flow.jpg", ["flofile.flo"], "diffused_input1.jpg", "diffused_blended_2.jpg", 'wraped.jpg', "blended2.jpg"
 ####################################
 # Bonus: Creating GIFs of predicted flows
 # ---------------------------------------