example radio

app.py

@@ -57,8 +57,8 @@ def label_to_color_image(label):
 
 # Gradio 인터페이스 정의
 iface = gr.Interface(
-    fn=lambda image: predict_segmentation(image, model),
-    inputs="image",
+    fn=predict_segmentation,
+    inputs=gr.Image(shape=(564,846)),
     outputs="image",
     examples=["city1.jpg","city2.jpg","city3.jpg"],
 )