Update app.py

app.py

@@ -26,13 +26,9 @@ if torch.cuda.is_available():
 def pix2pix(
     prompt,
     image,
-    text_guidance_scale,
-    image_guidance_scale,
     steps,
-    neg_prompt="",
-    width=512,
-    height=512,
-    seed=0,
+    image_guidance_scale,
+    seed
 ):
     print(psutil.virtual_memory())  # print memory usage
 
@@ -48,12 +44,12 @@ def pix2pix(
 
         result = pipe(
             prompt,
-            negative_prompt=neg_prompt,
             image=image,
             num_inference_steps=int(steps),
             image_guidance_scale=image_guidance_scale,
             guidance_scale=text_guidance_scale,
             generator=generator,
+            negative_prompt=neg_prompt,
         )
 
         # return replace_nsfw_images(result)
@@ -125,9 +121,9 @@ def infer(prompt,video_in, seed_in, trim_value):
     print("set stop frames to: " + str(n_frame))
     
     for i in frames_list[0:int(n_frame)]:
-        pix2pix_img = pix2pix(prompt,i,5.5,1.5,15,"",512,512,seed_in)
+        pix2pix_img = pix2pix(prompt,i,5.5,15,seed_in)
         print(pix2pix_img)
-        image = pix2pix_img
+        image = PIL.Image.open(pix2pix_img)
         rgb_im = image.convert("RGB")
   
         # exporting the image