Update app.py
Browse files
app.py
CHANGED
@@ -127,7 +127,9 @@ def infer():
|
|
127 |
pil2diff_img = Image.open("./frame1.jpg")
|
128 |
diffused_img = generate_images(pil2diff_img, prompt)
|
129 |
print(f"DIFFUSED IMG: {diffused_img[1]}")
|
130 |
-
diffused_img =
|
|
|
|
|
131 |
input_frame_1 = read_image(str("input1.jpg"), ImageReadMode.UNCHANGED)
|
132 |
print(f"FRAME 1: {input_frame_1}")
|
133 |
input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
|
|
|
127 |
pil2diff_img = Image.open("./frame1.jpg")
|
128 |
diffused_img = generate_images(pil2diff_img, prompt)
|
129 |
print(f"DIFFUSED IMG: {diffused_img[1]}")
|
130 |
+
diffused_img = Image.open(diffused_img)
|
131 |
+
diffused_img.save("input1.jpg")
|
132 |
+
|
133 |
input_frame_1 = read_image(str("input1.jpg"), ImageReadMode.UNCHANGED)
|
134 |
print(f"FRAME 1: {input_frame_1}")
|
135 |
input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
|