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)
|