Update app.py

app.py

@@ -1,4 +1,4 @@
-import scipy
+from scipy.special import softmax
 import streamlit as st
 import torch
 import numpy as np
@@ -39,7 +39,7 @@ text = title + '. ' + abstract
 tokens_info = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
 out_first = model_first(**tokens_info).pooler_output
 out_second = model_second(out_first).detach().numpy()
-out_second = scipy.special.softmax(out_second)
+out_second = softmax(out_second)
 indices = np.argsort(out_second)[0][::-1]
 sum_prob = 0
 for i in indices: