File size: 724 Bytes
0ee4111 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# app.py
import gradio as gr
from handler import EndpointHandler
# 初始化你的 ONNX 模型
handler = EndpointHandler(path=".")
def api_call(image, parameters):
data = {
"inputs": image, # PIL.Image
"parameters": parameters or {}
}
result = handler(data)[0]
return result # JSON only（含 base64 mask）
demo = gr.Interface(
fn=api_call,
inputs=[
gr.Image(type="pil", label="image"),
gr.JSON(label="parameters (point_coords, point_labels, return_mask_image)")
],
outputs=gr.JSON(label="result"),
title="Edge SAM API",
description="Pure API Space — send image + parameters, get mask JSON."
)
if __name__ == "__main__":
demo.launch()
|