Update src/model_def.py

src/model_def.py

@@ -2,8 +2,6 @@ import torch
 import torch.nn as nn
 import numpy as np
 
-# Positional Encoding class
-def class (rename to PositionalEncoding)
 class PositionalEncoding(nn.Module):
     def __init__(self, d_model, max_len=32):
         super().__init__()
@@ -17,8 +15,6 @@ class PositionalEncoding(nn.Module):
     def forward(self, x):
         return x + self.pe[:, :x.size(1)].to(x.device)
 
-# Transformer emotion classifier
-def class (rename to EmotionTransformer)
 class EmotionTransformer(nn.Module):
     def __init__(self, vocab_size, embed_dim=64, num_heads=4, num_classes=None):
         super().__init__()