Update app.py
Browse files
app.py
CHANGED
@@ -11,10 +11,6 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
|
|
11 |
"nvidia/segformer-b1-finetuned-cityscapes-1024-1024")
|
12 |
|
13 |
|
14 |
-
|
15 |
-
# 모델 입력 크기를 확인합니다.
|
16 |
-
input_size = model.input_shape[1:3]
|
17 |
-
|
18 |
# 모델 예측 함수를 정의합니다.
|
19 |
def classify_image(img):
|
20 |
# 이미지를 모델 입력 크기에 맞게 조정합니다.
|
|
|
11 |
"nvidia/segformer-b1-finetuned-cityscapes-1024-1024")
|
12 |
|
13 |
|
|
|
|
|
|
|
|
|
14 |
# 모델 예측 함수를 정의합니다.
|
15 |
def classify_image(img):
|
16 |
# 이미지를 모델 입력 크기에 맞게 조정합니다.
|