Update app.py

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
-from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
+from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
 from PIL import Image
 import torch
 
-# 모델과 피처 추출기 로드
+# 모델과 이미지 프로세서 로드
 model_name = "nvidia/mit-b0"
-feature_extractor = SegformerFeatureExtractor.from_pretrained(model_name)
+image_processor = SegformerImageProcessor.from_pretrained(model_name)
 model = SegformerForSemanticSegmentation.from_pretrained(model_name)
 
 def remove_background(image):
     # 이미지에서 피처 추출
-    inputs = feature_extractor(images=image, return_tensors="pt")
+    inputs = image_processor(images=image, return_tensors="pt")
     outputs = model(**inputs)
     logits = outputs.logits
     # 결과에서 가장 가능성 높은 클래스 선택
@@ -31,7 +31,7 @@ def remove_background(image):
     return masked_image
 
 # Gradio 인터페이스 정의
-iface = gr.Interface(fn=remove_background, inputs=gr.inputs.Image(type="pil"), outputs="image", title="Background Remover", description="Upload an image to remove the background.")
+iface = gr.Interface(fn=remove_background, inputs=gr.Image(type="pil"), outputs="image", title="Background Remover", description="Upload an image to remove the background.")
 
 # 앱 실행
 iface.launch()