Update app.py

app.py

@@ -127,10 +127,10 @@ def infer(prompt,video_in, seed_in, trim_value):
     for i in frames_list[0:int(n_frame)]:
         pix2pix_img = pix2pix(prompt,5.5,1.5,i,15,"",512,512,seed_in)
         image = pix2pix_img
-        rgb_im = image.convert("RGB")
+        #rgb_im = image.convert("RGB")
   
         # exporting the image
-        rgb_im.save(f"result_img-{i}.jpg")
+        image.save(f"result_img-{i}.jpg")
         result_frames.append(f"result_img-{i}.jpg")
         print("frame " + i + "/" + str(n_frame) + ": done;")