Spaces:
Runtime error
Runtime error
dashasabirova
commited on
Commit
•
6eab1ff
1
Parent(s):
7537bb1
Update app.py
Browse files
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:
|