Spaces:
Sleeping
Sleeping
File size: 905 Bytes
03eec87 ca2592c 03eec87 65977ce 6373819 65977ce 03eec87 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
from setuptools import setup, find_packages
setup(
name='commafixer',
version='0.1.0',
description='Fixing commas using Deep Learning.',
author='Karol Lasocki',
author_email='karolasocki@gmail.com',
url='https://huggingface.co/spaces/klasocki/comma-fixer',
packages=find_packages(include=['commafixer', 'commafixer.*']),
install_requires=[
"fastapi == 0.101.1",
"uvicorn == 0.23.2",
"nltk == 3.8.1",
'peft==0.5.0',
"torch == 2.0.1",
"transformers == 4.31.0",
# for the tokenizer of the baseline model
"protobuf == 4.24.0",
"sentencepiece == 0.1.99",
],
extras_require={
'training': [
'datasets==2.14.4',
'notebook',
'seqeval',
'evaluate==0.4.0'
],
'test': [
'pytest',
'httpx'
]
},
)
|