Spaces:

malmukhtar
/

Accessibility-labeler

Sleeping

malmukhtar commited on Oct 8, 2023

Commit

ced4e42

1 Parent(s): 9b30f59

Update accessibility_classifier/classifier/model.py

Files changed (1) hide show

accessibility_classifier/classifier/model.py CHANGED Viewed

@@ -98,8 +98,8 @@ class Model:
         all_predictions = F.softmax(
             probabilities.logits, dim=1).cpu().numpy().flatten()
-        accessibility_prediction = all_predictions[1]
-        nonaccessibility_prediction = all_predictions[0]
         return (accessibility_prediction, nonaccessibility_prediction)

         all_predictions = F.softmax(
             probabilities.logits, dim=1).cpu().numpy().flatten()
+        accessibility_prediction = all_predictions[0]
+        nonaccessibility_prediction = all_predictions[1]
         return (accessibility_prediction, nonaccessibility_prediction)