Update utils/xlnet_model.py

utils/xlnet_model.py

@@ -24,13 +24,14 @@ class XLNetAnswerAssessmentModel(nn.Module):
         hidden = 768
         self.fc1 = nn.Linear(hidden, 256)
         self.fc2 = nn.Linear(256, 64)
-        self.out = nn.Linear(64, 1)
+        self.output = nn.Linear(64, 1)  # ← Change from `self.out` to `self.output`
 
     def forward(self, input_ids, attention_mask=None):
         pooled = self.xlnet(input_ids, attention_mask).last_hidden_state.mean(1)
         x = torch.relu(self.fc1(pooled))
         x = torch.relu(self.fc2(x))
-        return torch.sigmoid(self.out(x))
+        return torch.sigmoid(self.output(x))  # ← And change here too
+
 
 # Initialize model and tokenizer
 xlnet_available = False