app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,11 @@ import torch
|
|
| 6 |
|
| 7 |
# ---------------- 配置模型列表 ----------------
|
| 8 |
MODEL_LIST = [
|
|
|
|
| 9 |
"yangy50/garbage-classification",
|
|
|
|
| 10 |
"ahmzakif/TrashNet-Classification",
|
|
|
|
| 11 |
"harriskr14/trashnet-vit"
|
| 12 |
]
|
| 13 |
|
|
@@ -58,7 +61,7 @@ iface = gr.Interface(
|
|
| 58 |
fn=classify_image,
|
| 59 |
inputs=gr.Image(type="pil", label="上传图片"),
|
| 60 |
outputs=[gr.Textbox(label="所有模型预测结果")],
|
| 61 |
-
title="
|
| 62 |
description="上传图片后，每个模型独立输出预测结果，不做任何人工干预。"
|
| 63 |
)
|
| 64 |
|
|
|
|
| 6 |
|
| 7 |
# ---------------- 配置模型列表 ----------------
|
| 8 |
MODEL_LIST = [
|
| 9 |
+
"prithivMLmods/Trash-Net",
|
| 10 |
"yangy50/garbage-classification",
|
| 11 |
+
"eunoiawiira-vgg-realwaste-classification",
|
| 12 |
"ahmzakif/TrashNet-Classification",
|
| 13 |
+
"ee8225-group4-vit-trashnet-enhanced",
|
| 14 |
"harriskr14/trashnet-vit"
|
| 15 |
]
|
| 16 |
|
|
|
|
| 61 |
fn=classify_image,
|
| 62 |
inputs=gr.Image(type="pil", label="上传图片"),
|
| 63 |
outputs=[gr.Textbox(label="所有模型预测结果")],
|
| 64 |
+
title="垃圾分类全模型检测",
|
| 65 |
description="上传图片后，每个模型独立输出预测结果，不做任何人工干预。"
|
| 66 |
)
|
| 67 |
|