Ahsen Khaliq commited on
Commit
1d090f4
1 Parent(s): 257cf7d

Update app.py

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