581630c4e0
- 增加模型加载的异常处理，当 pipeline 直接加载失败时回退为
手动加载 AutoImageProcessor 和 AutoModelForImageClassification
- 添加 torchvision 依赖
21 lines
500 B
TOML
21 lines
500 B
TOML
[project]
|
|
name = "banana-disease-leaf"
|
|
version = "0.1.0"
|
|
description = "香蕉叶病害智能识别系统 - 基于 ViT 视觉变换器的香蕉叶片病害分类应用"
|
|
readme = "README.md"
|
|
requires-python = ">=3.14"
|
|
dependencies = [
|
|
"numpy>=2.3.5",
|
|
"plotly>=6.5.0",
|
|
"ruff>=0.14.8",
|
|
"streamlit==1.52.1",
|
|
"torch>=2.7.0",
|
|
"torchvision>=0.22.0",
|
|
"transformers>=4.52.0",
|
|
"pillow>=11.2.0",
|
|
]
|
|
|
|
[[tool.uv.index]]
|
|
url = "https://mirrors.aliyun.com/pypi/simple"
|
|
default = true
|