Update app.py

app.py

@@ -40,11 +40,14 @@ def process_image(image, prompt):
     mask_image = cv2.cvtColor(np.array(mask_image), cv2.COLOR_RGB2GRAY)
 
     # Perform inpainting with LAMA
-    input_dict = {"image": image, "mask": mask_image}
-    inpainted_image = lama_model.inference(data=input_dict)["data"][0]
+    # input_dict = {"image": image, "mask": mask_image}
+    # imageio.imwrite("./data/data_mask.png", input_dict["mask"])
+    # os.system('python predict.py model.path=/home/user/app/ indir=/home/user/app/data/ outdir=/home/user/app/dataout/ device=cpu')
+    # # inpainted_image = lama_model.inference(data=input_dict)["data"][0]
     
-    inpainted_image = cv2.cvtColor(inpainted_image, cv2.COLOR_BGR2RGB)
-    inpainted_image = Image.fromarray(inpainted_image)
+    inpainted_image = Image.new('RGB', (image.shape[1], image.shape[0]), (0, 0, 0))
+    # inpainted_image = cv2.cvtColor(inpainted_image, cv2.COLOR_BGR2RGB)
+    # inpainted_image = Image.fromarray(inpainted_image)
 
     return mask_image, inpainted_image