[tool.poetry]
name = "genai"
version = "0.1.0"
description = ""
authors = ["Samuel"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.10"
torch = {version = "^2.2.1+cu118", source = "pytorch"}
torchvision = {version = "^0.17.1+cu118", source = "pytorch"}
torchaudio = {version = "^2.2.1+cu118", source = "pytorch"}
transformers = "^4.38.1"
langchain = "^0.1.9"
gradio = "^4.19.2"


[[tool.poetry.source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu118"
priority = "explicit"


[tool.poetry.group.dev.dependencies]
ruff = "^0.2.2"
pytest = "^8.0.1"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"