Update app.py

app.py

@@ -57,17 +57,11 @@ def pix2pix(
         )
 
         # return replace_nsfw_images(result)
-        return result.images, result.nsfw_content_detected, seed
+        return result[0]
     except Exception as e:
-        return None, None, error_str(e)
+        return None
+
 
-def error_str(error, title="Error"):
-    return (
-        f"""#### {title}
-            {error}"""
-        if error
-        else ""
-    )
 
 def get_frames(video_in):
     frames = []
@@ -132,8 +126,8 @@ 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)
-        images = pix2pix_img[0]
-        rgb_im = images[0].convert("RGB")
+        image = pix2pix_img
+        rgb_im = image.convert("RGB")
   
         # exporting the image
         rgb_im.save(f"result_img-{i}.jpg")