Update app.py

app.py

@@ -128,9 +128,9 @@ def infer():
     diffused_img = generate_images(pil2diff_img, prompt)
     print(f"DIFFUSED IMG: {diffused_img[1]}")
     
-    diffused_img[1].save("input1.jpg")
+    diffused_img[1].save("diffused_input1.jpg")
     
-    input_frame_1 = read_image(str("input1.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("./frame2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
@@ -265,7 +265,7 @@ def infer():
     blend2 = Image.blend(res,blend2,0.5)
     blend2.save("blended2.jpg")
     
-    return "done", "predicted_flow.jpg", ["flofile.flo"], 'raw_frame2.jpg', 'wraped.jpg', "blended2.jpg"
+    return "done", "predicted_flow.jpg", ["flofile.flo"], "diffused_input1.jpg", 'raw_frame2.jpg', 'wraped.jpg', "blended2.jpg"
 ####################################
 # Bonus: Creating GIFs of predicted flows
 # ---------------------------------------
@@ -294,4 +294,4 @@ def infer():
 # ffmpeg -f image2 -framerate 30 -i predicted_flow_%d.jpg -loop -1 flow.gif
 
       
-gr.Interface(fn=infer, inputs=[], outputs=[gr.Textbox(), gr.Image(), gr.Files(), gr.Image(), gr.Image(), gr.Image()]).launch()
+gr.Interface(fn=infer, inputs=[], outputs=[gr.Textbox(), gr.Image(label="flow"), gr.Files(), gr.Image(label="diffused input"), gr.Image(), gr.Image(label="wraped flow to img2"), gr.Image(label="blended result to diffuse")]).launch()