Sure
this is from pytorch_model.py
from torch import nn 
import torch.nn.functional as F



class Model(nn.Module):
    def __init__(self):
        super(Model, self).__init__()
        self.linearA = nn.Linear(13,100)
        self.linearB = nn.Linear(100, 50)
        self.linearC = nn.Linear(50, 1)

    def forward(self, x):
        yA = F.relu(self.linearA(x))
        yB = F.relu(self.linearB(yA))
        return self.linearC(yB)

and this is the part from utils.py
data.model = Model()
data.model.load_state_dict(torch.load(data.model_path))