pip install transformers | |
import streamlit as st | |
from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
model_name = "peace4ever/roberta-large-finetuned-mongolian_v4" | |
tokenizer = AutoTokenizer.from_pretrained(model_name) | |
model = AutoModelForSequenceClassification.from_pretrained(model_name) | |
text = st.text_area("Өгүүлбэр оруулна уу?") | |
encoded_input = tokenizer(text, return_tensors="pt") | |
output = model(**encoded_input) | |
st.json(output) | |