nitinbhayana commited on
Commit
71e03f9
·
verified ·
1 Parent(s): ed2c954

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,6 +76,7 @@ def submit_title():
76
  ner = ner_title(title)
77
  st.session_state.ner_dict = ner
78
  st.session_state.submitted_title = True
 
79
 
80
 
81
  def submit_ner_keywords():
@@ -147,7 +148,7 @@ def ner_title(title):
147
  #output = re.sub(' ": "', '": "', output)
148
 
149
  #return eval(pipe(text)[0]["generated_text"].split("### NER Response:\n")[-1])
150
- return convert_to_dictionary((output[0]['generated_text'].split("### NER Response:\n")[-1]))
151
 
152
  def reset_app():
153
  # Clear relevant session states
 
76
  ner = ner_title(title)
77
  st.session_state.ner_dict = ner
78
  st.session_state.submitted_title = True
79
+ st.write(ner)
80
 
81
 
82
  def submit_ner_keywords():
 
148
  #output = re.sub(' ": "', '": "', output)
149
 
150
  #return eval(pipe(text)[0]["generated_text"].split("### NER Response:\n")[-1])
151
+ return output# convert_to_dictionary((output[0]['generated_text'].split("### NER Response:\n")[-1]))
152
 
153
  def reset_app():
154
  # Clear relevant session states