wrapper228 commited on
Commit
6d77f28
1 Parent(s): ffa287c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ def predict_topic_by_title_and_abstract(text, model):
15
  with torch.no_grad():
16
  logits = model(**tokens).logits
17
  probs = torch.nn.functional.softmax(logits[0], dim=0).numpy() * 100
18
- ans = list(zip(probs,labels.values()))
 
19
  ans.sort(reverse=True)
20
  sum = 0
21
  i = 0
 
15
  with torch.no_grad():
16
  logits = model(**tokens).logits
17
  probs = torch.nn.functional.softmax(logits[0], dim=0).numpy() * 100
18
+ #ans = list(zip(probs,labels.values()))
19
+ ans = list(zip(probs,[x for x in range(8)]))
20
  ans.sort(reverse=True)
21
  sum = 0
22
  i = 0