File size: 932 Bytes
6d9ae67
d74ac86
 
 
 
 
 
05bae7f
6d9ae67
05bae7f
d74ac86
f0689a3
 
 
14f73ef
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import os
from pathlib import Path

import streamlit as st

x = st.slider('Select a value')
st.write(x, 'squared is', x * x)
# from huggingface_hub import login

# login(os.getenv('HF_TOKEN'))

# hub_folder = Path('~/.cache/huggingface/hub')
# for path in hub_folder.walk():
#     st.write(path)

from transformers import AutoModelForSequenceClassification
from transformers import AutoTokenizer
from transformers import TextClassificationPipeline

model = AutoModelForSequenceClassification.from_pretrained(
    "issai/rembert-sentiment-analysis-polarity-classification-kazakh")
tokenizer = AutoTokenizer.from_pretrained("issai/rembert-sentiment-analysis-polarity-classification-kazakh")

pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)

reviews = ["Бұл бейнефильм маған түк ұнамады.", "Осы кітап қызық сияқты."]

for review in reviews:
    st.write(pipe(review))