fffiloni commited on
Commit
dd7adab
1 Parent(s): fa67c68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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 = diffused_img.save("input1.jpg")
 
 
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)