Update app.py
Browse files
app.py
CHANGED
@@ -50,11 +50,11 @@ def process(title, summary):
|
|
50 |
probs = torch.exp(out[0])
|
51 |
sorted_indexes = torch.argsort(probs, descending=True)
|
52 |
probs_sum = idx = 0
|
53 |
-
|
54 |
while probs_sum < 0.95:
|
55 |
prob_idx = sorted_indexes[idx]
|
56 |
prob = probs[prob_idx]
|
57 |
-
|
58 |
idx += 1
|
59 |
probs_sum += prob
|
60 |
return str
|
@@ -63,4 +63,5 @@ title = st.text_area("Title", height=30)
|
|
63 |
|
64 |
summary = st.text_area("Summary", height=200)
|
65 |
|
66 |
-
|
|
|
|
50 |
probs = torch.exp(out[0])
|
51 |
sorted_indexes = torch.argsort(probs, descending=True)
|
52 |
probs_sum = idx = 0
|
53 |
+
res = []
|
54 |
while probs_sum < 0.95:
|
55 |
prob_idx = sorted_indexes[idx]
|
56 |
prob = probs[prob_idx]
|
57 |
+
res.append(f'{classes[prob_idx]}: {prob:.3f}')
|
58 |
idx += 1
|
59 |
probs_sum += prob
|
60 |
return str
|
|
|
63 |
|
64 |
summary = st.text_area("Summary", height=200)
|
65 |
|
66 |
+
for sting in process(title, summary):
|
67 |
+
st.markdown(string)
|