ui test

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 import os
+from main_infer import INFER_API
+
 
 def process_image(image):
     # 设置保存图片的路径，指向当前工程的 images 目录
@@ -15,17 +17,19 @@ def process_image(image):
     
     # 保存图片
     image.save(image_path)
-    
-    return f"图片已保存至 {image_path}"  # 返回保存的位置信息
+    infer_api = INFER_API()
+    score = infer_api.test(img_path)
+
+    return f"图片已保存至 {image_path},得分是：{score}"  # 返回保存的位置信息
 
 # 定义输入：一个图片选择框
-inputs = gr.inputs.Image(label="Upload Image", type="pil")
+image_input = gr.components.Image(label="Upload Image", type="pil")
 
 # 定义输出：一个文本框，用来显示确认信息或处理结果
-outputs = gr.outputs.Textbox()
+text_output = gr.components.Textbox()
 
 # 创建 Interface 对象，设置 live=False 以添加提交按钮
-demo = gr.Interface(fn=process_image, inputs=inputs, outputs=outputs, live=False)
+demo = gr.Interface(fn=process_image, inputs=image_input, outputs=text_output, live=False)
 
 # 启动界面
 demo.launch()

requirements.txt

@@ -16,4 +16,4 @@ uvicorn==0.22.0
 pydantic==1.10.9
 torch==1.13.1
 torchvision==0.14.1
-
+gradio