import torch
from torch import nn


class TicketClassifier(nn.Module):
    def __init__(self):
        super().__init__()
        self.layers = nn.Sequential(
            nn.Linear(4, 3),
            nn.Dropout(p=0.5),
        )

    def forward(self, inputs):
        return self.layers(inputs)


class_names = ["standard", "priority", "urgent"]
device = torch.device("cpu")
model = TicketClassifier().to(device)