sentiments

language_models_project/app.py

@@ -1,5 +1,3 @@
-#from main import classify
-#import easyocr as ocr  #OCR
 import streamlit as st  #Web App
 from main import classify
 
@@ -16,8 +14,12 @@ input_sentences = st.text_area("Sentences", value="", height=200)
 
 data = input_sentences.split('\n')
 
+#if st.button("Classify"):
 for i in data:
-    st.write(f"{i} :: Classification - {classify(i)}")
+    j = classify(i)[0]
+    sentiment = j['label'] == 'POS'
+    confidence = j['score']
+    st.write(f"{i} :: Classification - {'positive' if sentiment else 'negative'} with confidence {confidence}")
  
 
 st.markdown("Link to the app - [image-to-text-app on 🤗 Spaces](https://huggingface.co/spaces/Amrrs/image-to-text-app)")