Update app.py

app.py

@@ -32,11 +32,14 @@ user_query = st_tags(
 # Add selectbox in streamlit
 option1 = st.sidebar.selectbox(
      'Which transformers model would you like to be selected?',
-     ('multi-qa-MiniLM-L6-cos-v1'))
+     ('multi-qa-MiniLM-L6-cos-v1','null','null'))
 
 option2 = st.sidebar.selectbox(
      'Which corss-encoder model would you like to be selected?',
-     ('cross-encoder/ms-marco-MiniLM-L-6-v2'))
+     ('cross-encoder/ms-marco-MiniLM-L-6-v2','null','null'))
+
+search=pipeline(option)
+st.sidebar.success("Load Successfully!")
 
 if not torch.cuda.is_available():
     print("Warning: No GPU found. Please add GPU to your notebook")
@@ -130,4 +133,5 @@ def search(query):
 
 st.write("## Results:")
 if st.button('Generate Sentence'):
-    out = search(query = user_query)
+    out = search(query = user_query)
+    st.success(out)