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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,7 +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
- st.write("ner:",ner)
80
 
81
  def submit_ner_keywords():
82
  st.write("Title:", st.session_state.title)
@@ -147,7 +147,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 output#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
+
80
 
81
  def submit_ner_keywords():
82
  st.write("Title:", st.session_state.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