Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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)
|