Update app.py

app.py

@@ -63,7 +63,7 @@ def CCA_Analysis(orig_image,predict_image,erode_iteration,open_iteration):
         cnts = cnts[0]
         c_area = cv2.contourArea(cnts)
         # threshhold for tooth count
-        if c_area>1000:
+        if c_area>2000:
             count2+=1
 
         (x,y),radius = cv2.minEnclosingCircle(cnts)
@@ -167,6 +167,11 @@ if image_file is not None:
     img_dc=convert_one_channel(img)
     decay_mask = detect_decays_static_th(img_dc, predict1)
     
+    
+    predict1 = cv2.cvtColor(predict1,cv2.COLOR_GRAY2RGB)
+    img_dc = cv2.cvtColor(img_dc,cv2.COLOR_GRAY2RGB)
+    decay_mask = cv2.cvtColor(decay_mask,cv2.COLOR_GRAY2RGB)
+    
     mask = np.uint8(predict1 * 255)
     _, mask = cv2.threshold(mask, thresh=255/2, maxval=255, type=cv2.THRESH_BINARY)
     cnts, hierarchy = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)