Update app.py

app.py

@@ -13,7 +13,6 @@ def object_detection(image):
     outputs = model(**inputs)
 
     # 결과 처리 로직을 구현하세요.
-    # 예제 코드는 결과 처리 부분을 단순화하고 있습니다.
     # 실제로는 탐지된 객체를 이미지 위에 박스로 그리는 등의 처리가 필요할 수 있습니다.
     # 여기서는 outputs의 요약된 정보만을 반환합니다.
     return outputs.logits.tolist()
@@ -21,8 +20,8 @@ def object_detection(image):
 # Gradio 인터페이스 설정 및 앱 실행
 iface = gr.Interface(
     fn=object_detection,
-    inputs=gr.Image(type="pil", tool="editor"),
-    outputs=gr.Text(),
+    inputs=gr.Image(shape=(512, 512)),
+    outputs="json",
     title="Object Detection with DETR",
     description="Upload an image, and the model will detect objects in the image."
 )