README.md 604 B

一个基于 Python 的 AI 项目, 依赖 CUDA, 通过在代码中加载 AI 模型和调用 ollama 提供的 AI 模型, 提供了以下功能:

  • 图像理解 > moondream
  • 语音识别 > funasr
  • 文本分析 > qwen2.5:1.5b
  • 文本翻译 > qwen2.5:1.5b

依赖

  • cuda:12.4

导出依赖

pip freeze > requirements.txt

pip freeze 会导出当前环境下所有安装的包

安装依赖

pip install -r requirements.txt

构建

项目根目录下运行此命令:

docker build -t pyai:12345678 .

运行

项目根目录下运行此命令:

docker compose up -d