Update app.py

app.py

@@ -46,7 +46,7 @@ def predict(comment_sentence):
         outputs = model(ids, mask, token_type_ids)
 
     print(outputs)
-    fin_outputs.extend(torch.sigmoid(torch.tensor(outputs)).cpu().detach().numpy().tolist())
+    fin_outputs.extend(torch.sigmoid(outputs[0]).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