import streamlit as st
from model import classify_text


st.markdown("### Sentiment classification (negative vs. positive)")
title = st.text_area("Your sentiment for classification", "2010 So I spent a few hours making something for fun. . . If you don't know I am a HUGE RhandlerR fan and Maya is one of my favorite characters. So I decided to make myself a wallpaper for my PC. . Here is the original image versus the creation I made :) Enjoy! pic.twitter.com/mLsI5wf9Jg")
if st.button("Classify!"):
    prob = float(classify_text(title))
    st.markdown(f"**Model**: {round(prob, 5)} probability of being positive")