import streamlit as st
from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification

model_name = "hiieu/Meta-Llama-3-8B-Instruct-function-calling-json-mode"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)

pipe = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
text = st.text_area("Enter your text:")

if text:
    out = pipe(text)
    st.json(out)