Update app.py

app.py

@@ -76,7 +76,7 @@ def return_num_columns(img):
     return num_col
 
 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":
+    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
@@ -130,7 +130,7 @@ def return_scaled_image(img, num_col, width_early, model_name):
             img_w_new = width_early
             img_h_new = int(img.shape[0] / float(img.shape[1]) * width_early)
         img_new = resize_image(img, img_h_new, img_w_new)
-    elif model_name=="SBB/eynollah-main-regions":
+    elif model_name=="SBB/eynollah-main-regions" or "SBB/eynollah-textline_light":
         if num_col == 1:
             img_w_new = 1000
             img_h_new = int(img.shape[0] / float(img.shape[1]) * img_w_new)