KazSandra / app.py
lgblkb's picture
refactor: remove login
05bae7f
raw
history blame
932 Bytes
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))