Update app.py

app.py

@@ -74,6 +74,7 @@ def draw_plot(pred_img, seg):
     ax.tick_params(width=0.0, labelsize=25)
     return fig
 
+
 def sepia(input_img):
     input_img = Image.fromarray(input_img)
 
@@ -82,9 +83,12 @@ def sepia(input_img):
     logits = outputs.logits
 
     logits = tf.transpose(logits, [0, 2, 3, 1])
+
+    # 크기 조정 코드 추가
     logits = tf.image.resize(
-        logits, input_img.size[::-1]
-    )  # We reverse the shape of `image` because `image.size` returns width and height.
+        logits, [input_img.size[1], input_img.size[0]]
+    )
+
     seg = tf.math.argmax(logits, axis=-1)[0]
 
     color_seg = np.zeros(