initial commit

app.py

@@ -145,7 +145,8 @@ if st.button('Analyze'):
             data['context']=paragraph
             print(data)
 
-            predictions = run_prediction(data['question'], data['context'], 'akdeniz27/roberta-base-cuad', n_best_size=number_results )
+            predictions = run_prediction(data['question'], data['context'], 'akdeniz27/roberta-base-cuad',
+                                         n_best_size=int(number_results))
             # print(resp)
             # predictions=resp.json()
             # print(predictions)

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 int(n_best_size) > 1:
+    if n_best_size > 1:
         output_nbest_file = "nbest.json"
 
     timer = time.time()