File size: 1,661 Bytes
049266b 601925f 049266b 601925f 049266b 601925f 049266b 601925f 049266b 601925f 049266b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import time
import streamlit as st
from flair.data import Sentence
from flair.models import SequenceTagger
from flair.visual.ner_html import render_ner_html
checkpoints = [
"qanastek/pos-french",
]
@st.cache(suppress_st_warning=True, allow_output_mutation=True)
def get_model(model_name):
# Load the model
return SequenceTagger.load(model_name)
def main():
st.title("🥖 French-Part-Of-Speech-Tagging")
checkpoint = st.selectbox("Choose model", checkpoints)
model = get_model(checkpoint)
default_text = "George Washington est allé à Washington"
input_text = st.text_area(
label="Original text",
value=default_text,
)
start = None
if st.button("🧠 Compute"):
start = time.time()
with st.spinner("Search for Part-Of-Speech Tags 🔍"):
# Build Sentence
s = Sentence(input_text)
# predict tags
model.predict(s)
# # print predicted pos tags
# result = sentence.to_tagged_string()
try:
st.write(render_ner_html(sentences=[s], wrap_page=False), unsafe_allow_html=True)
except Exception as e:
st.error("Some error occured!" + str(e))
st.stop()
st.write("---")
st.markdown(
"Built by [Yanis Labrak](https://www.linkedin.com/in/yanis-labrak-8a7412145/) 🚀"
)
st.markdown(
"_Source code made with [FlairNLP](https://github.com/flairNLP/flair)_"
)
if start is not None:
st.text(f"prediction took {time.time() - start:.2f}s")
if __name__ == "__main__":
main()
|