import streamlit as st
from transformers import pipeline

# Load model from Hugging Face
classifier = pipeline("text-classification", model="Soundaryasos/Verdictclassifications")

# UI
st.title("Criminal Case Sentencing Predictor")
st.write("Enter case details to predict the likely outcome.")

# User input
user_input = st.text_area("Case Description:")

if st.button("Predict"):
    result = classifier(user_input)
    st.write("Predicted Outcome:", result[0]['label'])