Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from predictor import inference | |
| def index_predict(img): | |
| outputs, preds, heights, bust, waist, hips, description = inference(img, epoch = 7) | |
| return heights, round(float(bust)), round(float(waist)), round(float(hips)), description[0], description[1] | |
| with gr.Blocks() as demo: | |
| gr.Markdown( | |
| """ | |
| # 身材數據評估器 - Body Index Predictor | |
| ### Input A FACE and get the body index | |
| """ | |
| ) | |
| with gr.Row(): | |
| image = gr.Image(type="pil") | |
| with gr.Column(scale=1, min_width=600): | |
| # 設定輸出元件 | |
| heights = gr.Textbox(label="Heignt(身高)") | |
| bust = gr.Textbox(label="Bust(胸圍)") | |
| waist = gr.Textbox(label="Waist(腰圍)") | |
| hips = gr.Textbox(label="Hips(臀圍)") | |
| en_des = gr.Textbox(label="English description") | |
| zh_des = gr.Textbox(label="Chinese description") | |
| #設定按鈕 | |
| submit = gr.Button("Submit") | |
| #設定按鈕點選事件 | |
| submit.click(fn=index_predict, inputs=image, outputs=[heights, bust, waist, hips, en_des, zh_des]) | |
| demo.launch() |