Update app.py

app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 import numpy as np
 from PIL import Image
-from tensorflow.keras.models import load_model
+import tensorflow as tf
+from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
-# 이미지 분류 모델을 불러옵니다. 여기에서는 간단한 예시 모델을 사용합니다.
-model = load_model('nvidia/segformer-b1-finetuned-cityscapes-1024-1024')  # 모델 경로를 적절하게 수정하세요.
+feature_extractor = SegformerFeatureExtractor.from_pretrained(
+    "nvidia/segformer-b1-finetuned-cityscapes-1024-1024"
 
 # 모델 입력 크기를 확인합니다.
 input_size = model.input_shape[1:3]