peace4ever's picture
Update app.py
e8c85cb verified
raw
history blame
No virus
481 Bytes
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)