app.py

import streamlit as st
from transformers import pipeline

# Load your model
model = pipeline("text-classification", model="KevSun/Personality_LM")

st.title("Personality Prediction App")

st.write("Enter your text below to predict personality traits:")

user_input = st.text_area("Your text here:")

if st.button("Predict"):
    if user_input:
        # Process the input and get predictions
        result = model(user_input)
        
        # Display results
        st.write("Predicted personality traits:")
        st.write(result)
    else:
        st.write("Please enter some text to analyze.")