Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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
|
| 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
|