Update app.py

app.py

@@ -65,8 +65,9 @@ try:
   top_k = 10
   print(top_k)
 
+  model_name = st.sidebar.selectbox(label='Select Model to Apply',  options=['ajitrajasekharan/biomedical', 'BERT'], index=0,  key = "model_name")
 
-  bert_tokenizer, bert_model  = load_bert_model('ajitrajasekharan/biomedical')
+  bert_tokenizer, bert_model  = load_bert_model(model_name)
   default_text = "Imatinib is used to treat"
   
  
@@ -84,7 +85,7 @@ try:
             
 
             try:
-                res = get_bert_prediction(default_text,top_k)
+                res = get_bert_prediction(input_text,top_k)
 	
 
                 st.header("JSON:")