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)) | |