Update app.py

app.py

@@ -1,40 +1,35 @@
-import cv2
+import gradio as gr
 import requests
 import tempfile
 import os
+
+import cv2
 import numpy as np
-import gradio as gr
 
 def resize_img(img):
-    # 表示したい画像サイズ
-    h_base, w_base = (315, 420)
-
-    # 読み込み画像サイズ
-    print('original: ', img.shape[:2])
-    h, w = img.shape[:2]
+    h_base, w_base = (315, 420) # 表示したい画像サイズ
+    h, w = img.shape[:2]        # 読み込み画像サイズ
     
-    # 画像サイズが違う画像はリサイズ
-    if h!=h_base and w!=w_base:
+    if h!=h_base and w!=w_base: # 画像サイズが違う画像はリサイズ
+        
         # 基本サイズが縦の方が小さいので縦に合わせるように倍率を縮小
         out_img = cv2.resize(img, None, None, h_base/h, h_base/h)
 
         r_w = out_img.shape[1] # リサイズ後の横幅
-        # 横サイズが大きい場合はトリミング
-        if r_w > w_base:
-            out_img = out_img[:,int((r_w-w_base)/2):int((r_w+w_base)/2)]
         
-        # 横サイズが小さい場合は白で埋める
-        elif r_w < w_base:
+        if r_w > w_base: # 横サイズが大きい場合はトリミング
+            out_img = out_img[:,int((r_w-w_base)/2):int((r_w+w_base)/2)]
+            
+        elif r_w < w_base: # 横サイズが小さい場合は白で埋める
             tmp = out_img
             # 白で埋める
             out_img = np.uint8(np.array([[[255]*3]*w_base]*h_base))
             out_img[:,int((w_base-r_w)/2):int((w_base+r_w)/2)] = tmp
-    # 画像サイズが同じ場合はスルー 
-    else:
+
+    else: # 画像サイズが同じ場合はスルー 
         out_img = img
-    
-    print('resize: ',out_img.shape[:2])
-    
+
+    # RGBに変換
     rgb_img = cv2.cvtColor(out_img, cv2.COLOR_BGR2RGB)
 
     return rgb_img