vahidrezanezhad commited on
Commit
99ca461
1 Parent(s): 9c79a29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,7 +76,8 @@ def return_num_columns(img):
76
  return num_col
77
 
78
  def return_scaled_image(img, num_col, width_early, model_name):
79
- if model_name == ("SBB/eynollah-main-regions-aug-rotation" or "SBB/eynollah-main-regions-aug-scaling" or "SBB/eynollah-main-regions-ensembled"):
 
80
  if num_col == 1 and width_early < 1100:
81
  img_w_new = 2000
82
  img_h_new = int(img.shape[0] / float(img.shape[1]) * 2000)
 
76
  return num_col
77
 
78
  def return_scaled_image(img, num_col, width_early, model_name):
79
+ if model_name == "SBB/eynollah-main-regions-aug-rotation" or "SBB/eynollah-main-regions-aug-scaling" or "SBB/eynollah-main-regions-ensembled":
80
+ print('here')
81
  if num_col == 1 and width_early < 1100:
82
  img_w_new = 2000
83
  img_h_new = int(img.shape[0] / float(img.shape[1]) * 2000)