Update app.py

app.py

@@ -57,10 +57,12 @@ monuseg_inf = MoNuSegInference(
 
 def click_process(image_input , type_dataset):
     if type_dataset == "pannuke":
+        if image_input.shape[0] > 512 and image_input.shape[1] > 512:
+            image_input = cv2.resize(image_input, (512,512))
         pannuke_inf.run_single_image_inference(pannuke_model,image_input)
     else:
-        resize_shape = (512,512)
-        image_input = cv2.resize(image_input, resize_shape)
+        if image_input.shape[0] > 512 and image_input.shape[1] > 512:
+            image_input = cv2.resize(image_input, (512,512))
         monuseg_inf.run_single_image_inference(monuseg_inf.model, image_input)
 
     image_output = cv2.imread("raw_pred.png")