Ahsen Khaliq commited on
Commit
8a6d741
1 Parent(s): 7909c4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def infer(img):
26
  im = Image.fromarray(result[0]['mask'])
27
  im.save("./data/data_mask.png")
28
  os.system("ls data")
29
- os.system('python predict.py model.path=/home/user/app/big-lama/ indir=./data outdir=./dataout device=cpu')
30
  os.system("ls dataout")
31
  return "./dataout/data_mask.png"
32
  inputs = gr.inputs.Image(type='pil', label="Original Image")
 
26
  im = Image.fromarray(result[0]['mask'])
27
  im.save("./data/data_mask.png")
28
  os.system("ls data")
29
+ os.system('python predict.py model.path=/home/user/app/big-lama/ indir=/home/user/app/data/ outdir=/home/user/app/dataout/ device=cpu')
30
  os.system("ls dataout")
31
  return "./dataout/data_mask.png"
32
  inputs = gr.inputs.Image(type='pil', label="Original Image")