nitinbhayana commited on
Commit
9e7c2eb
·
verified ·
1 Parent(s): 3bc5038

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -47,7 +47,8 @@ def ner_title(title):
47
  # "inputs": prompt,
48
  # })
49
 
50
- return eval(pipe(prompt)[0]["generated_text"].split("### NER Response:\n")[-1])
 
51
  #return(eval(output[0]['generated_text'].split("### NER Response:\n")[-1]))
52
  # def ner_title(title):
53
  # word_list = title.split()
@@ -61,6 +62,7 @@ def submit_title():
61
  ner = ner_title(title)
62
  st.session_state.submitted_title = True
63
  st.session_state.ner_dict = ner
 
64
 
65
  def submit_ner_keywords():
66
  st.subheader("Product Features:")
 
47
  # "inputs": prompt,
48
  # })
49
 
50
+ return (pipe(prompt)[0]["generated_text"])
51
+ #return eval(pipe(prompt)[0]["generated_text"].split("### NER Response:\n")[-1])
52
  #return(eval(output[0]['generated_text'].split("### NER Response:\n")[-1]))
53
  # def ner_title(title):
54
  # word_list = title.split()
 
62
  ner = ner_title(title)
63
  st.session_state.submitted_title = True
64
  st.session_state.ner_dict = ner
65
+ st.write("ner:",ner)
66
 
67
  def submit_ner_keywords():
68
  st.subheader("Product Features:")