Update app.py

app.py

@@ -3,7 +3,6 @@ from transformers import AutoImageProcessor, BeitForSemanticSegmentation
 from PIL import Image
 import torch
 import numpy as np
-import matplotlib.pyplot as plt
 
 # 모델과 이미지 프로세서를 로드합니다.
 processor = AutoImageProcessor.from_pretrained("microsoft/beit-base-finetuned-ade-640-640")
@@ -34,8 +33,8 @@ def predict(image):
 # Gradio 인터페이스 설정
 gr_interface = gr.Interface(
     fn=predict,
-    inputs=gr.inputs.Image(type="pil"),
-    outputs="image",
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
     title="Image Segmentation with BEiT",
     description="Upload an image to perform segmentation using the microsoft/beit-base-finetuned-ade-640-640 model."
 )