saad noor commited on
Commit ·
cab47c8
1
Parent(s): 604ea5c
launched app
Browse files
app.py
CHANGED
|
@@ -213,7 +213,7 @@ configs['image'] = {
|
|
| 213 |
|
| 214 |
def evaluate(img_path, model=general_model, img_model=image_model,\
|
| 215 |
configs=configs, flags=flags):
|
| 216 |
-
print('starting')
|
| 217 |
img = cv2.imread(img_path)
|
| 218 |
res = get_masks(img, general_model, image_model, flags, configs)
|
| 219 |
if res['status']==-1:
|
|
@@ -231,7 +231,7 @@ def evaluate(img_path, model=general_model, img_model=image_model,\
|
|
| 231 |
}
|
| 232 |
for i, mask in enumerate(masks):
|
| 233 |
img = overlay(image=img, mask=mask, color=color_map[i], alpha=0.4)
|
| 234 |
-
print('finishing')
|
| 235 |
return img
|
| 236 |
|
| 237 |
# output = evaluate(img_path=sample_path, model=general_model, img_model=image_model,\
|
|
@@ -251,4 +251,4 @@ interface_image = gr.Interface(
|
|
| 251 |
title="Document Layout Segmentor",
|
| 252 |
examples=sample_path,
|
| 253 |
cache_examples=True,
|
| 254 |
-
)
|
|
|
|
| 213 |
|
| 214 |
def evaluate(img_path, model=general_model, img_model=image_model,\
|
| 215 |
configs=configs, flags=flags):
|
| 216 |
+
# print('starting')
|
| 217 |
img = cv2.imread(img_path)
|
| 218 |
res = get_masks(img, general_model, image_model, flags, configs)
|
| 219 |
if res['status']==-1:
|
|
|
|
| 231 |
}
|
| 232 |
for i, mask in enumerate(masks):
|
| 233 |
img = overlay(image=img, mask=mask, color=color_map[i], alpha=0.4)
|
| 234 |
+
# print('finishing')
|
| 235 |
return img
|
| 236 |
|
| 237 |
# output = evaluate(img_path=sample_path, model=general_model, img_model=image_model,\
|
|
|
|
| 251 |
title="Document Layout Segmentor",
|
| 252 |
examples=sample_path,
|
| 253 |
cache_examples=True,
|
| 254 |
+
).launch()
|