Update app.py

app.py

@@ -1,23 +1,23 @@
-import streamlit as st
-import pandas as pd
-from rnc_morphemer.NeuralMorphemeSegmentation.neural_morph_segm import load_cls
-
-path = 'rnc_morphemer/models/morphodict_10_07_2023.json'
-def predict(lemma):
-    model = load_cls(path)
-
-    labels, _ = model._predict_probs([lemma])[0]
-    morphemes, morpheme_types = model.labels_to_morphemes(
-        lemma, labels, return_probs=False, return_types=True
-    )
-
-    parsing = [
-        {"morpheme": morpheme, "type": morpheme_type}
-        for morpheme, morpheme_type in zip(morphemes, morpheme_types)
-    ]
-
-    return parsing
-
-input = st.text_input(label='Морфемный разбор слова:')
-st.write(pd.DataFrame(predict(input)))
-
+import streamlit as st
+import pandas as pd
+from rnc_morphemer.NeuralMorphemeSegmentation.neural_morph_segm import load_cls
+
+path = 'rnc_morphemer/models/morphodict_10_07_2023.json'
+def predict(lemma):
+    model = load_cls(path)
+
+    labels, _ = model._predict_probs([lemma])[0]
+    morphemes, morpheme_types = model.labels_to_morphemes(
+        lemma, labels, return_probs=False, return_types=True
+    )
+
+    parsing = [
+        {"morpheme": morpheme, "type": morpheme_type}
+        for morpheme, morpheme_type in zip(morphemes, morpheme_types)
+    ]
+
+    return parsing
+st.header('Слово на морфемы онлайн')
+input = st.text_input(label='Морфемный разбор слова:')
+st.write(pd.DataFrame(predict(input)))
+