initial commit

app.py

@@ -74,7 +74,8 @@ with open('data/'+contracts_files[idx]) as f:
     contract_data = f.read()
 
 # upload contract
-user_upload = st.sidebar.file_uploader('Please upload your own', type=['docx', 'pdf', 'txt'], accept_multiple_files=False)
+user_upload = st.sidebar.file_uploader('Please upload your own', type=['docx', 'pdf', 'txt'],
+                                       accept_multiple_files=False)
 print(user_upload)
 
 # process upload
@@ -130,17 +131,8 @@ if st.button('Analyze'):
     if (not len(paragraph)==0) and not (len(question)==0):
         print('getting predictions')
         with st.spinner(text='Analysis in progress...'):
-            #predictions = run_prediction([question], paragraph, '../models/roberta-base/')
-            data = {}
-            data['question'] = [question]
-            data['context'] = paragraph
-            print(data)
-
-            predictions = run_prediction(data['question'], data['context'], 'akdeniz27/roberta-base-cuad',
+            predictions = run_prediction([question], paragraph, 'akdeniz27/roberta-base-cuad',
                                          n_best_size=int(number_results))
-            # print(resp)
-            # predictions=resp.json()
-            # print(predictions)
         if predictions['0'] == "":
             answer = 'No answer found in document'
         else:
@@ -148,9 +140,9 @@ if st.button('Analyze'):
                 answer = predictions['0']
                 st.text_area(label="Answer", value=f"{answer}")
             else:
-                f = open("nbest.json")
-                st.success(f.readlines())
+                with open("nbest.json") as jf:
+                    data = json.load(jf)
+                    st.success(data["0"])
             st.success("Successfully processed contract!")
-            st.success(question)
     else:
         st.write("Unable to call model, please select question and contract")