| import gradio as gr | |
| from PIL import Image | |
| from infer import infer | |
| MODEL = None | |
| LOCAL_REPO_PATH = None | |
| def process_image(image): | |
| """调用 infer() 进行处理并返回结果""" | |
| global MODEL, LOCAL_REPO_PATH | |
| ret, MODEL, LOCAL_REPO_PATH= infer(image, MODEL, LOCAL_REPO_PATH) | |
| return ret | |
| iface = gr.Interface( | |
| fn=process_image, | |
| inputs=gr.Image(type="pil"), | |
| outputs=gr.Image(type="numpy"), | |
| title="RGB手部姿态估计（手部mesh重建正在建设中……）", | |
| description="上传或拍摄一张右手单手的RGB图片，输出手部姿态估计结果。" | |
| ) | |
| if __name__ == "__main__": | |
| iface.launch(debug=True) | |