TangYiJay commited on
Commit
606734b
·
verified ·
1 Parent(s): f768864
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -16,6 +16,7 @@ MODEL_LIST = [
16
  models = []
17
  processors = []
18
  devices = []
 
19
  print("🔹 正在加载模型,请稍等...")
20
 
21
  for model_name in MODEL_LIST:
@@ -49,13 +50,11 @@ def classify_image(image: Image.Image):
49
  except Exception as e:
50
  results[model_name] = f"error: {e}"
51
 
52
- # 格式化输出
53
  results_text = "\n".join([f"{name}: {label}" for name, label in results.items()])
54
 
55
  # 计算最终标签(投票法)
56
  valid_labels = [lbl for lbl in results.values() if not lbl.startswith("error")]
57
  final_label = Counter(valid_labels).most_common(1)[0][0] if valid_labels else "Unknown"
58
-
59
  results_text += f"\n\n最终标签: {final_label}"
60
  return results_text
61
 
 
16
  models = []
17
  processors = []
18
  devices = []
19
+
20
  print("🔹 正在加载模型,请稍等...")
21
 
22
  for model_name in MODEL_LIST:
 
50
  except Exception as e:
51
  results[model_name] = f"error: {e}"
52
 
 
53
  results_text = "\n".join([f"{name}: {label}" for name, label in results.items()])
54
 
55
  # 计算最终标签(投票法)
56
  valid_labels = [lbl for lbl in results.values() if not lbl.startswith("error")]
57
  final_label = Counter(valid_labels).most_common(1)[0][0] if valid_labels else "Unknown"
 
58
  results_text += f"\n\n最终标签: {final_label}"
59
  return results_text
60