Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -59,6 +59,8 @@ def click_process(image_input , type_dataset):
|
|
59 |
if type_dataset == "pannuke":
|
60 |
pannuke_inf.run_single_image_inference(pannuke_model,image_input)
|
61 |
else:
|
|
|
|
|
62 |
monuseg_inf.run_single_image_inference(monuseg_inf.model, image_input)
|
63 |
|
64 |
image_output = cv2.imread("pred_img.png")
|
|
|
59 |
if type_dataset == "pannuke":
|
60 |
pannuke_inf.run_single_image_inference(pannuke_model,image_input)
|
61 |
else:
|
62 |
+
resize_shape = (512,512)
|
63 |
+
image_input = cv2.resize(image_input, resize_shape)
|
64 |
monuseg_inf.run_single_image_inference(monuseg_inf.model, image_input)
|
65 |
|
66 |
image_output = cv2.imread("pred_img.png")
|