Update app.py

app.py

@@ -42,7 +42,6 @@ def visualize_model_output(prediction, img):
     output = np.zeros(prediction.shape)
 
     for unq_class in unique_classes:
-        print(unq_class,'unq_class')
         rgb_class_unique = rgb_colors[str(int(unq_class))]
         output[:,:,0][prediction[:,:,0]==unq_class] = rgb_class_unique[0]
         output[:,:,1][prediction[:,:,0]==unq_class] = rgb_class_unique[1]
@@ -77,7 +76,6 @@ def return_num_columns(img):
 
 def return_scaled_image(img, num_col, width_early, model_name):
     if model_name == "SBB/eynollah-main-regions-aug-rotation" or "SBB/eynollah-main-regions-aug-scaling" or "SBB/eynollah-main-regions-ensembled" or "SBB/eynollah-textline":
-        print('here')
         if num_col == 1 and width_early < 1100:
             img_w_new = 2000
             img_h_new = int(img.shape[0] / float(img.shape[1]) * 2000)