initial commit

app.py

@@ -136,16 +136,16 @@ if st.button('Analyze'):
         if predictions['0'] == "":
             answer = 'No answer found in document'
         else:
-            if number_results == '1':
-                answer = f"Answer: {predictions['0']}"
-                # st.text_area(label="Answer", value=f"{answer}")
-            else:
-                answer = ""
-                with open("nbest.json") as jf:
-                    data = json.load(jf)
-                    for i in range(int(number_results)):
-                        answer += f"Answer {i+1}: {data['0'][i]['text']} -- \n"
-                        answer += f"Probability: {round(data['0'][i]['probability']*100,1)}%\n\n"
+            # if number_results == '1':
+            #     answer = f"Answer: {predictions['0']}"
+            #     # st.text_area(label="Answer", value=f"{answer}")
+            # else:
+            answer = ""
+            with open("nbest.json") as jf:
+                data = json.load(jf)
+                for i in range(int(number_results)):
+                    answer += f"Answer {i+1}: {data['0'][i]['text']} -- \n"
+                    answer += f"Probability: {round(data['0'][i]['probability']*100,1)}%\n\n"
             st.success(answer)
             # st.success("Successfully processed contract!")
     else:

predict.py

@@ -102,7 +102,7 @@ def run_prediction(question_texts, context_text, model_path, n_best_size=1):
     print(all_results)
 
     output_nbest_file = None
-    if n_best_size > 1:
+    if n_best_size) > 1:
         output_nbest_file = "nbest.json"
 
     timer = time.time()