Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,8 @@ def ner_title(title):
|
|
47 |
# "inputs": prompt,
|
48 |
# })
|
49 |
|
50 |
-
return
|
|
|
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:")
|