DeepShcool / src /config.py
AndreiBar's picture
Update src/config.py
84c79bf verified
Raw
History Blame Contribute Delete
926 Bytes
#from pathlib import Path
from typing import Optional, Union
#from pydantic import Field
from pydantic_settings import BaseSettings
class Config(BaseSettings):
# имя модели и задачи подсмотрите в карточке модели
# https://huggingface.co/ProsusAI/finbert
# Use this model -> Transformers
# и заполните эти переменные
model_name: Optional[str] = "ProsusAI/finbert"
task: Optional[str] = "text-classification"
# Для задания с датасетом заполните свой username с HF и придумайте имя для датасета
#hf_user: Optional[str] = "AndreiBar"
#hf_dataset_name: Optional[str] = "UploadTestDataset"
#save_directory: Union[Path, str] = PROJECT_PATH / "output"
#path_to_example: Union[Path, str] = PROJECT_PATH / "text_example" / "example.txt"
config = Config()