Heiko Hotz commited on
Commit
0a89e6e
1 Parent(s): e557652

initial commit

Browse files
Files changed (2) hide show
  1. app.py +10 -10
  2. predict.py +1 -1
app.py CHANGED
@@ -136,16 +136,16 @@ if st.button('Analyze'):
136
  if predictions['0'] == "":
137
  answer = 'No answer found in document'
138
  else:
139
- if number_results == '1':
140
- answer = f"Answer: {predictions['0']}"
141
- # st.text_area(label="Answer", value=f"{answer}")
142
- else:
143
- answer = ""
144
- with open("nbest.json") as jf:
145
- data = json.load(jf)
146
- for i in range(int(number_results)):
147
- answer += f"Answer {i+1}: {data['0'][i]['text']} -- \n"
148
- answer += f"Probability: {round(data['0'][i]['probability']*100,1)}%\n\n"
149
  st.success(answer)
150
  # st.success("Successfully processed contract!")
151
  else:
 
136
  if predictions['0'] == "":
137
  answer = 'No answer found in document'
138
  else:
139
+ # if number_results == '1':
140
+ # answer = f"Answer: {predictions['0']}"
141
+ # # st.text_area(label="Answer", value=f"{answer}")
142
+ # else:
143
+ answer = ""
144
+ with open("nbest.json") as jf:
145
+ data = json.load(jf)
146
+ for i in range(int(number_results)):
147
+ answer += f"Answer {i+1}: {data['0'][i]['text']} -- \n"
148
+ answer += f"Probability: {round(data['0'][i]['probability']*100,1)}%\n\n"
149
  st.success(answer)
150
  # st.success("Successfully processed contract!")
151
  else:
predict.py CHANGED
@@ -102,7 +102,7 @@ def run_prediction(question_texts, context_text, model_path, n_best_size=1):
102
  print(all_results)
103
 
104
  output_nbest_file = None
105
- if n_best_size > 1:
106
  output_nbest_file = "nbest.json"
107
 
108
  timer = time.time()
 
102
  print(all_results)
103
 
104
  output_nbest_file = None
105
+ if n_best_size) > 1:
106
  output_nbest_file = "nbest.json"
107
 
108
  timer = time.time()