Soundaryasos's picture
Update app.py
97070f7 verified
raw
history blame contribute delete
472 Bytes
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'])