Update app.py

app.py

@@ -44,8 +44,9 @@ def predict(comment_sentence):
     # Forward pass through the model
     with torch.no_grad():
         outputs = model(ids, mask, token_type_ids)
-    
-    fin_outputs.extend(torch.sigmoid(torch.tensor(outputs.logits)).cpu().detach().numpy().tolist())
+
+    print(outputs)
+    fin_outputs.extend(torch.sigmoid(torch.tensor(outputs)).cpu().detach().numpy().tolist())
     print(fin_outputs)
     # Convert the outputs to boolean values based on the threshold
     outputs_boolean = np.array(fin_outputs) >= 0.25