Update app.py

app.py

@@ -24,6 +24,8 @@ fdic = {
     "weight": "bold"
 }
 
+threshold = 80  # 置信度阈值
+
 label_color_dict = {}
 
 def query_data(in_pil_img: Image.Image):
@@ -57,7 +59,7 @@ def get_annotated_image(in_pil_img):
         box = prediction['box']
         label = prediction['label']
         score = round(prediction['score'] * 100, 1)
-        if score < 50:
+        if score < threshold:
             continue  # 过滤掉低置信度的预测结果
 
         if label not in label_color_dict:  # 为每个类别随机分配颜色, 后续维持一致
@@ -72,7 +74,7 @@ def get_annotated_image(in_pil_img):
         font = ImageFont.truetype(font="arial.ttf", size=font_size)  # 确保你有可用的字体文件
 
         # 获取文本边界框
-        text = f"{label}: {score}%"
+        text = f"{label}: {score}%" 
         text_bbox = draw.textbbox((0, 0), text, font=font)
 
         # 绘制矩形