feat: add form input
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from pathlib import Path
|
2 |
|
3 |
import streamlit as st
|
@@ -8,7 +9,7 @@ from transformers import TextClassificationPipeline
|
|
8 |
|
9 |
|
10 |
@st.cache_data()
|
11 |
-
def
|
12 |
model = AutoModelForSequenceClassification.from_pretrained(
|
13 |
"issai/rembert-sentiment-analysis-polarity-classification-kazakh")
|
14 |
tokenizer = AutoTokenizer.from_pretrained("issai/rembert-sentiment-analysis-polarity-classification-kazakh")
|
@@ -22,10 +23,16 @@ assert static_folder.exists()
|
|
22 |
st.write((static_folder / 'description.txt').read_text())
|
23 |
st.image(str(static_folder / 'kazsandra.jpg'))
|
24 |
|
25 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
# reviews = ["Бұл бейнефильм маған түк ұнамады.", "Осы кітап қызық сияқты."]
|
27 |
-
pipe = get_pipe()
|
28 |
-
# for review in reviews:
|
29 |
-
if input_text:
|
30 |
-
out = pipe(input_text)[0]
|
31 |
-
st.text("Label: {label}\nScore: {score}".format(**out))
|
|
|
1 |
+
from operator import setitem
|
2 |
from pathlib import Path
|
3 |
|
4 |
import streamlit as st
|
|
|
9 |
|
10 |
|
11 |
@st.cache_data()
|
12 |
+
def load_model():
|
13 |
model = AutoModelForSequenceClassification.from_pretrained(
|
14 |
"issai/rembert-sentiment-analysis-polarity-classification-kazakh")
|
15 |
tokenizer = AutoTokenizer.from_pretrained("issai/rembert-sentiment-analysis-polarity-classification-kazakh")
|
|
|
23 |
st.write((static_folder / 'description.txt').read_text())
|
24 |
st.image(str(static_folder / 'kazsandra.jpg'))
|
25 |
|
26 |
+
pipe = load_model()
|
27 |
+
|
28 |
+
with st.form('main_form'):
|
29 |
+
input_text = st.text_area('Input text', placeholder='Provide your text, e.g. "Осы кітап қызық сияқты".')
|
30 |
+
is_submitted = st.form_submit_button(label='Submit')
|
31 |
+
if is_submitted:
|
32 |
+
if input_text:
|
33 |
+
out = pipe(input_text)[0]
|
34 |
+
st.text("Label: {label}\nScore: {score}".format(**out))
|
35 |
+
else:
|
36 |
+
st.text("Please provide your text first.")
|
37 |
+
|
38 |
# reviews = ["Бұл бейнефильм маған түк ұнамады.", "Осы кітап қызық сияқты."]
|
|
|
|
|
|
|
|
|
|