downgrade streamlit

app.py

@@ -1,5 +1,6 @@
 import pandas as pd
 import streamlit as st
+import streamlit.components.v1 as components
 from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
 
 from src import GBRT, wikipedia_search, google_search
@@ -67,9 +68,9 @@ def get_candidates(mentions_tags):
 def display_tag(text, typ, label):
     if label != 'NIL':
         label = "https://en.wikipedia.org/wiki/" + label
-        return f'<a style="margin: 0px 5px 0px 0px; padding: 1px 0px 2px 4px; border-radius: 4px; text-decoration:none; background-color:{COLOR[typ]}; color: white; cursor:pointer; white-space: nowrap; font-weight:400" href={label} target="_blank"><span style="margin-right:2px;">{text}</span><span style="background-color: white; color: {COLOR[typ]}; padding: 1px 2px; border-radius: 4px; font-size: 0.8em">{typ}</span></a>'
+        return f'<a style="margin: 0px 5px 0px 0px; padding: 2px 4px; border-radius: 4px; text-decoration:none; background-color:{COLOR[typ]}; color: white; cursor:pointer; white-space: nowrap; font-weight:400" href={label} target="_blank"><span style="margin-right:2px;">{text}</span><span style="background-color: white; color: {COLOR[typ]}; padding: 1px 2px; border-radius: 4px; font-size: 0.8em">{typ}</span></a>'
     else:
-        return f'<span style="margin: 0px 5px 0px 0px; padding: 1px 0px 2px 4px; border-radius: 4px; text-decoration:none; background-color: #EF4444; color: white; cursor:not-allowed; white-space: nowrap;"><span style="margin-right:2px">{text}</span><span style="background-color: white; color: #EF4444; padding: 1px 2px; border-radius: 4px; font-size: 0.8em">{typ}</span></span>'
+        return f'<span style="margin: 0px 5px 0px 0px; padding: 2px 4px; border-radius: 4px; text-decoration:none; background-color: #EF4444; color: white; cursor:not-allowed; white-space: nowrap;"><span style="margin-right:2px">{text}</span><span style="background-color: white; color: #EF4444; padding: 1px 2px; border-radius: 4px; font-size: 0.8em">{typ}</span></span>'
 
 
 
@@ -96,7 +97,8 @@ def main(text):
 
     with col2:
         st.write("### Disambiguated Text")
-        st.write(tagged, unsafe_allow_html=True)
+        components.html(f'<div style="white-space: pre-line; line-height: 1.65; font-family: sans-serif">{tagged}</div>', 
+                        height=700, scrolling=True)
 
     df = pd.DataFrame(data=preditions, columns=['Mention', 'Prediction', 'Confidence'])
     st.write("**Additional Information**")

requirements.txt

@@ -4,6 +4,5 @@ pandas
 wikipedia2vec
 scikit-learn
 transformers
-streamlit==1.8.1
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch