Update app.py

app.py

@@ -70,10 +70,10 @@ def magic_lama(img):
     inverted_mask = ImageOps.invert(mask)
     
     
-    imageio.imwrite(f"./data/data_{i}.png", img)
-    imageio.imwrite(f"./data/data_mask_{i}.png", inverted_mask)
+    imageio.imwrite(f"./data/data.png", img)
+    imageio.imwrite(f"./data/data_mask.png", inverted_mask)
     os.system('python predict.py model.path=/home/user/app/ indir=/home/user/app/data/ outdir=/home/user/app/dataout/ device=cpu')
-    return f"./dataout/data_mask_{i}.png"
+    return f"./dataout/data_mask.png"
 
 def infer(video_in):
     # 1. break video into frames and get FPS
@@ -93,7 +93,9 @@ def infer(video_in):
     
     for i in frames_list[0:int(n_frame)]:
         lama_frame = magic_lama(i)
-        result_frames.append(lama_frame)
+        lama_frame = Image.open(lama_frame)
+        imageio.imwrite(f"cleaned_frame_{i}", lama_frame)
+        result_frames.append(f"cleaned_frame_{i}")
         print("frame " + i + "/" + str(n_frame) + ": done;")