app.py
Browse files
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 |
|