Spaces:

rrevoid
/

article_classifier

Runtime error

rrevoid commited on Mar 28, 2022

Commit

52297f7

1 Parent(s): 33fcbd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ cats = ["Computer Science", "Economics", "Electrical Engineering",
 def predict(outputs):
     top = 0
-    temp = 50000
     apr_probs = torch.nn.functional.softmax(torch.tensor([39253., 84., 220., 2263., 1214., 909., 66., 10661.]) / temp, dim=0)
     probs = nn.functional.softmax(outputs / apr_probs, dim=1).tolist()[0]

 def predict(outputs):
     top = 0
+    temp = 100000
     apr_probs = torch.nn.functional.softmax(torch.tensor([39253., 84., 220., 2263., 1214., 909., 66., 10661.]) / temp, dim=0)
     probs = nn.functional.softmax(outputs / apr_probs, dim=1).tolist()[0]