Update app.py

app.py

@@ -50,11 +50,11 @@ def process(title, summary):
     probs = torch.exp(out[0])
     sorted_indexes = torch.argsort(probs, descending=True)
     probs_sum = idx = 0
-    str = ''
+    res = []
     while probs_sum < 0.95:
         prob_idx = sorted_indexes[idx]
         prob = probs[prob_idx]
-        str += f'{classes[prob_idx]}: {prob:.3f}'+'<br>'    
+        res.append(f'{classes[prob_idx]}: {prob:.3f}')    
         idx += 1
         probs_sum += prob
     return str
@@ -63,4 +63,5 @@ title = st.text_area("Title", height=30)
 
 summary = st.text_area("Summary", height=200)
 
-st.markdown(f"{process(title, summary)}")
+for sting in process(title, summary):
+    st.markdown(string)