sebastianschramm

import torch
from litserve import LitAPI, LitServer
from pydantic import BaseModel, conint
from pydantic_settings import BaseSettings
from transformers import AutoModelForSequenceClassification, AutoTokenizer


class ToxicitySettings(BaseSettings):
    model_id: str = "s-nlp/roberta_toxicity_classifier"
    port: conint(ge=1024, le=65535) = 8000  # type: ignore