import gradio as gr import os from main_infer import INFER_API def process_image(image): # 设置保存图片的路径,指向当前工程的 images 目录 save_path = os.path.join(os.getcwd(), "images") # 确保目录存在 os.makedirs(save_path, exist_ok=True) # 为每个上传的图片生成唯一的文件名 from datetime import datetime timestamp = datetime.now().strftime("%Y%m%d%H%M%S") image_filename = f"uploaded_{timestamp}.png" image_path = os.path.join(save_path, image_filename) # 保存图片 image.save(image_path) infer_api = INFER_API() score = infer_api.test(image_path) return score # 返回保存的位置信息 # 定义输入:一个图片选择框 image_input = gr.components.Image(label="Upload Image", type="pil") # 定义输出:一个文本框,用来显示确认信息或处理结果 text_output = gr.components.Textbox() # 创建 Interface 对象,设置 live=False 以添加提交按钮 demo = gr.Interface(fn=process_image, inputs=image_input, outputs=text_output, live=False) # 启动界面 demo.launch()