sbert_punc_case_ru / setup.py
grazder's picture
safetensors update (#3)
eab0c50 verified
raw
history blame contribute delete
807 Bytes
from distutils.core import setup
setup(
name="sbert_punc_case_ru",
version="0.2",
description="Punctuation and Case Restoration model based on https://huggingface.co/sberbank-ai/sbert_large_nlu_ru",
author="Almira Murtazina",
author_email="ar.murtazina@skbkontur.ru",
packages=["sbert_punc_case_ru"],
install_requires=[
"transformers>=4.36.2",
"torch",
"numpy"
],
classifiers=[
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.6",
"Programming Language :: Python :: 3.7",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
)