fffiloni commited on
Commit
fb90370
1 Parent(s): 82002ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -135,7 +135,11 @@ def infer(prompt,video_in, seed_in, trim_value):
135
  for i in frames_list[0:int(n_frame)]:
136
  pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,seed_in)
137
  images = pix2pix_img[0]
138
- result_frames.append(images[0])
 
 
 
 
139
  print("frame " + i + ": done;")
140
 
141
  final_vid = create_video(result_frames, fps)
 
135
  for i in frames_list[0:int(n_frame)]:
136
  pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,seed_in)
137
  images = pix2pix_img[0]
138
+ rgb_im = images[0].convert("RGB")
139
+
140
+ # exporting the image
141
+ rgb_im.save("result_img.jpg")
142
+ result_frames.append("result_img.jpg")
143
  print("frame " + i + ": done;")
144
 
145
  final_vid = create_video(result_frames, fps)