from transformers import pipeline
import streamlit as st

generater = pipeline("text-generation", model="huggingtweets/cryptoanglio")
classifier = pipeline("text-classification", model="yzhangqs/Finace_news")

st.title("A Finance News Robot")
st.write("Classification for 3 classes: Positive, Negative and Neutral")

# Text input for user to enter the text to classify
text = st.text_area("Enter the text to start", "")
if st.button("Generate"):
  generated_text = generater(text)[0]['generated_text']
  result= classifier(generated_text)[0]
  score=result['score']
  label=result['label']


  st.write("Text:", generated_text)
  st.write("Label:", label)
  st.write("Score:", score)