Spaces:
Sleeping
Sleeping
Ahsen Khaliq
commited on
Commit
•
1d090f4
1
Parent(s):
257cf7d
Update app.py
Browse files
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
|
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")
|