Merge pull request #5 from VarshithaChennamsetti/milestone-3

patent_app.py

@@ -39,16 +39,20 @@ val_set = val_set.filter(lambda e: e['decision'] <= 1)
 # Display all patent numbers to select a file
 patent_num = st.selectbox("Select a patent based on its number",  val_set['patent_number'])
 
+button_1 = st.button('Get Data to predict!')
+
 # Get the abstract and claims data to predict
-if patent_num and st.button('Get Data to predict!'):
+if patent_num and button_1:
   # 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'])
-  claims_text = st.text_area('Abstract', val_set['claims'])
+  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 st.button('Predict!'):
+  if abstract_text and claims_text and button_2:
     # Model/tokenizer name or path to finetuned model
     model_name_or_path = './models/'
     model_name = 'distilbert-base-uncased'