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

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
  def submit_ner_keywords():
81
  st.write("Title:", st.session_state.title)
@@ -146,7 +147,7 @@ def ner_title(title):
146
  #output = re.sub(' ": "', '": "', output)
147
 
148
  #return eval(pipe(text)[0]["generated_text"].split("### NER Response:\n")[-1])
149
- return convert_to_dictionary((output[0]['generated_text'].split("### NER Response:\n")[-1]))
150
 
151
  def reset_app():
152
  # 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:",ner)
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 output#convert_to_dictionary((output[0]['generated_text'].split("### NER Response:\n")[-1]))
151
 
152
  def reset_app():
153
  # Clear relevant session states