Adding most likely topic

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import pandas as pd
+import numpy as np
 
 st.markdown('# Semantic search and topic classification (v1)')
 st.markdown(' - Author: hcontreras')
@@ -35,6 +36,14 @@ cos_scores = util.cos_sim(embedding_input_topic, embedding_sentence)[0][0]
 
 st.write('Score for topic', input_topic, ':', cos_scores.tolist())
 
+l_scores = []
+for topic in ['Space','Transportation','Health']:
+    embedding_input_topic = model.encode(topic)
+    cos_scores = util.cos_sim(embedding_input_topic, embedding_sentence)[0][0]
+    l_scores.append(cos_scores.tolist())
+
+st.write('Most likely topic:', ['Space','Transportation','Health'][np.argmax(l_scores)])
+
 st.markdown('## ')
 uploaded_file1 = st.file_uploader("Choose a file: sentence list")
 if uploaded_file1 is not None: