Hugging Face's logo

Spaces:
dashasabirova
/
HW-abstracts
Runtime error

App Files Community
dashasabirova commited on
Commit
6eab1ff
•
1 Parent(s): 7537bb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ text = title + '. ' + abstract
38
  tokens_info = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
39
  out_first = model_first(**tokens_info).pooler_output
40
  out_second = model_second(out_first).detach().numpy()
41
- out_second = softmax(out_second)
42
  indices = np.argsort(out_second)[0][::-1]
43
  sum_prob = 0
44
  for i in indices:
 
38
  tokens_info = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
39
  out_first = model_first(**tokens_info).pooler_output
40
  out_second = model_second(out_first).detach().numpy()
41
+ out_second = scipy.special.softmax(out_second)
42
  indices = np.argsort(out_second)[0][::-1]
43
  sum_prob = 0
44
  for i in indices: