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