Merge pull request #6 from VarshithaChennamsetti/milestone-3

patent_app.py

@@ -42,17 +42,15 @@ patent_num = st.selectbox("Select a patent based on its number",  val_set['paten
 button_1 = st.button('Get Data to predict!')
 
 # Get the abstract and claims data to predict
-if patent_num and button_1:
+if patent_num and (button_1 | st.button('Predict!')):
   # Display the abstract and claims
   val_set = val_set.filter(lambda e: e['patent_number'] == patent_num)
   
   abstract_text = st.text_area('Abstract', val_set['abstract'][0])
   claims_text = st.text_area('Claims', val_set['claims'][0])
   
-  button_2 = st.button('Predict!')
-
   # Predict on those texts   
-  if abstract_text and claims_text and button_2:
+  if abstract_text and claims_text and st.button('Predict!'):
     # Model/tokenizer name or path to finetuned model
     model_name_or_path = './models/'
     model_name = 'distilbert-base-uncased'