Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -258,11 +258,17 @@ selected_key = user_selection.split(" (")[0] # Remove the label part
|
|
| 258 |
selected_review = example_reviews[selected_key]
|
| 259 |
lower_case_review = selected_review.lower()
|
| 260 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 261 |
# Process the text when the button is clicked
|
| 262 |
if st.button("Process Review"):
|
| 263 |
if lower_case_review.strip():
|
| 264 |
bert_embedding = get_bert_embedding(lower_case_review)
|
| 265 |
-
spaBert_embedding, current_pseudo_sentences = processSpatialEntities(
|
| 266 |
combined_embedding = torch.cat((bert_embedding,spaBert_embedding),dim=-1)
|
| 267 |
|
| 268 |
if(dev_mode == True):
|
|
@@ -277,7 +283,7 @@ if st.button("Process Review"):
|
|
| 277 |
doc = nlp(selected_review)
|
| 278 |
|
| 279 |
# Highlight geo-entities with different colors
|
| 280 |
-
highlighted_text =
|
| 281 |
for ent in reversed(doc.ents):
|
| 282 |
if ent.label_ in COLOR_MAP:
|
| 283 |
color = COLOR_MAP[ent.label_][0]
|
|
|
|
| 258 |
selected_review = example_reviews[selected_key]
|
| 259 |
lower_case_review = selected_review.lower()
|
| 260 |
|
| 261 |
+
#Optional textbox for interactivity
|
| 262 |
+
user_input_review = st.text_area("Or type your own review here","")
|
| 263 |
+
|
| 264 |
+
review_to_process = user_input_review if user_input_review.strip() else selected_review
|
| 265 |
+
lower_case_review = review_to_process.lower()
|
| 266 |
+
|
| 267 |
# Process the text when the button is clicked
|
| 268 |
if st.button("Process Review"):
|
| 269 |
if lower_case_review.strip():
|
| 270 |
bert_embedding = get_bert_embedding(lower_case_review)
|
| 271 |
+
spaBert_embedding, current_pseudo_sentences = processSpatialEntities(review_to_process,nlp)
|
| 272 |
combined_embedding = torch.cat((bert_embedding,spaBert_embedding),dim=-1)
|
| 273 |
|
| 274 |
if(dev_mode == True):
|
|
|
|
| 283 |
doc = nlp(selected_review)
|
| 284 |
|
| 285 |
# Highlight geo-entities with different colors
|
| 286 |
+
highlighted_text = review_to_process
|
| 287 |
for ent in reversed(doc.ents):
|
| 288 |
if ent.label_ in COLOR_MAP:
|
| 289 |
color = COLOR_MAP[ent.label_][0]
|