Update app.py

app.py

@@ -37,14 +37,27 @@ def LoadModel():
 
 model = LoadModel()
 
+classes = ['Computer Science', 'Mathematics', 'Physics', 'Quantitative Biology', 'Statistics']
+
 def process(title, summary):
     text = title + summary
+    if not text.strip():
+        return ''
     model.eval()
     lines = [text]
     X = tokenizer(lines, padding=True, truncation=True, return_tensors="pt")
     out = model(X)
     probs = torch.exp(out[0])
-    return probs
+    sorted_indexes = torch.argsort(probs, descending=True)
+    probs_sum = idx = 0
+    str = ''
+    while probs_sum < 0.95:
+        prob_idx = sorted_indexes[idx]
+        prob = probs[prob_idx]
+        str += f'{classes[prob_idx]}: {prob:.3f}\n'    
+        idx += 1
+        probs_sum += prob
+    return str
     
 title = st.text_area("Title", height=30)