Update modeling.py

modeling.py

@@ -2,7 +2,6 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 from transformers import PreTrainedModel, AutoModel, AutoConfig
-from transformers.utils import ModelOutput
 
 class Encoder(nn.Module):
     def __init__(self, base_encoder):
@@ -45,7 +44,7 @@ class RobertaEmoPillars(PreTrainedModel):
         inputs = {"input_ids": input_ids, "attention_mask": attention_mask}
         emb = self.encoder(inputs)
         logits = self.classifier(emb)
-        return ModelOutput(logits=logits)
+        return logits
 
 
 # --- Registration for AutoModel ---