georad commited on
Commit
ebf3ad6
·
verified ·
1 Parent(s): 02c9abf

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +1 -1
pages/type_text.py CHANGED
@@ -138,7 +138,7 @@ pipe = load_pipe()
138
  dictA = {"Score": [], "SBS Code": [], "SBS Description V2.0": []}
139
  dfALL = pd.DataFrame.from_dict(dictA)
140
 
141
- if INTdesc_input is not None and createSBScodes_clicked == True:
142
  for i, result in enumerate(HF_model_results_displayed):
143
  dictA.update({"Score": "%.4f" % result[0]["score"], "SBS Code": df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[0]["corpus_id"]],"SBS_Code_Hyphenated"].values[0], "SBS Description V2.0": SBScorpus[result[0]["corpus_id"]]})
144
  dfALL = pd.concat([dfALL, pd.DataFrame([dictA])], ignore_index=True)
 
138
  dictA = {"Score": [], "SBS Code": [], "SBS Description V2.0": []}
139
  dfALL = pd.DataFrame.from_dict(dictA)
140
 
141
+ if INTdesc_input is not None and len(chapter_rows_indexes_list) >1 and createSBScodes_clicked == True:
142
  for i, result in enumerate(HF_model_results_displayed):
143
  dictA.update({"Score": "%.4f" % result[0]["score"], "SBS Code": df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[0]["corpus_id"]],"SBS_Code_Hyphenated"].values[0], "SBS Description V2.0": SBScorpus[result[0]["corpus_id"]]})
144
  dfALL = pd.concat([dfALL, pd.DataFrame([dictA])], ignore_index=True)