1

app.py

@@ -25,8 +25,8 @@ def predict(image):
 # Gradio 인터페이스 생성
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.inputs.Image(shape=(400, 600)),
-    outputs=gr.outputs.Image(),
+    inputs=gr.inputs.Image(shape=(400, 600), type='pil'),
+    outputs=gr.outputs.Image(type='pil'),
     examples=["image1.jpg", "image2.jpg", "image3.jpg"]  # 세 개의 예제 이미지 경로
 )