Update app.py

app.py

@@ -16,7 +16,7 @@ def hash_func(inp):
 
 @st.cache(allow_output_mutation=True, suppress_st_warning=True)
 def loading_models(model='roberta-base'):
-     return get_model(model), SentenceTransformer('all-MiniLM-L6-v2')
+     return get_model(model), SentenceTransformer('stsb-xlm-r-multilingual')#'all-MiniLM-L6-v2')
 
 @st.cache(allow_output_mutation=True, 
           suppress_st_warning=True,
@@ -74,7 +74,7 @@ The next token is predicted per probability and a weight if it is appeared in ke
         semantic_text = st.text_input("Enter users's history <Semantic> (optional, i.e., 'Microsoft' or 'President')", value="Microsoft")
         
         text = st.text_input("Enter a text for auto completion...", value='Where is Bill')
-        model = st.selectbox("Choose a model", ["roberta-base", "bert-base-uncased"])
+        model = st.selectbox("Choose a model", ["mmaguero/gn-bert-tiny-cased", "mmaguero/gn-bert-small-cased", "mmaguero/gn-bert-base-cased", "mmaguero/gn-bert-large-cased", "mmaguero/multilingual-bert-gn-base-cased", "mmaguero/beto-gn-base-cased"])
         
         data_load_state = st.text('1.Loading model ...')