KazSandra / app.py
lgblkb's picture
Download model during runtime
14f73ef
raw
history blame
922 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))