antoyo123 commited on
Commit
73fe085
1 Parent(s): af80700

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def process_image(image, prompt):
45
  imageio.imwrite("./data/data_mask.png", mask_image)
46
  os.system('python predict.py model.path=/home/user/app/ indir=/home/user/app/data/ outdir=/home/user/app/dataout/ device=cpu')
47
  # # inpainted_image = lama_model.inference(data=input_dict)["data"][0]
48
- inpainted_image = "./data/data_mask.png"
49
 
50
  # inpainted_image = Image.new('RGB', (image.shape[1], image.shape[0]), (0, 0, 0))
51
  # inpainted_image = cv2.cvtColor(inpainted_image, cv2.COLOR_BGR2RGB)
 
45
  imageio.imwrite("./data/data_mask.png", mask_image)
46
  os.system('python predict.py model.path=/home/user/app/ indir=/home/user/app/data/ outdir=/home/user/app/dataout/ device=cpu')
47
  # # inpainted_image = lama_model.inference(data=input_dict)["data"][0]
48
+ inpainted_image = "./dataout/data_mask.png"
49
 
50
  # inpainted_image = Image.new('RGB', (image.shape[1], image.shape[0]), (0, 0, 0))
51
  # inpainted_image = cv2.cvtColor(inpainted_image, cv2.COLOR_BGR2RGB)