Update app.py

app.py

@@ -132,25 +132,25 @@ def return_scaled_image(img, num_col, width_early, model_name):
     elif model_name=="SBB/eynollah-main-regions":
         if num_col == 1:
             img_w_new = 1000
-            img_h_new = int(img_org.shape[0] / float(img_org.shape[1]) * img_w_new)
+            img_h_new = int(img.shape[0] / float(img.shape[1]) * img_w_new)
             
         elif num_col == 2:
             img_w_new = 1500
-            img_h_new = int(img_org.shape[0] / float(img_org.shape[1]) * img_w_new)
+            img_h_new = int(img.shape[0] / float(img.shape[1]) * img_w_new)
             
         elif num_col == 3:
             img_w_new = 2000
-            img_h_new = int(img_org.shape[0] / float(img_org.shape[1]) * img_w_new)
+            img_h_new = int(img.shape[0] / float(img.shape[1]) * img_w_new)
             
         elif num_col == 4:
             img_w_new = 2500
-            img_h_new = int(img_org.shape[0] / float(img_org.shape[1]) * img_w_new)
+            img_h_new = int(img.shape[0] / float(img.shape[1]) * img_w_new)
         elif num_col == 5:
             img_w_new = 3000
-            img_h_new = int(img_org.shape[0] / float(img_org.shape[1]) * img_w_new)
+            img_h_new = int(img.shape[0] / float(img.shape[1]) * img_w_new)
         else:
             img_w_new = 4000
-            img_h_new = int(img_org.shape[0] / float(img_org.shape[1]) * img_w_new)
+            img_h_new = int(img.shape[0] / float(img.shape[1]) * img_w_new)
         img_resized = resize_image(img,img_h_new, img_w_new )
 
         img_new = otsu_copy_binary(img_resized)