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()
|