Update app.py

app.py

@@ -8,9 +8,9 @@ from transformers import pipeline
 
 # Load pipeline
 pipe_lr = joblib.load(open("emotion_detector_pipe_lr.pkl","rb"))
-emoroberta_emotion_classifier = pipeline(
-    "text-classification", model="arpanghoshal/EmoRoBERTa", return_all_scores=True
-)
+#emoroberta_emotion_classifier = pipeline(
+#    "text-classification", model="arpanghoshal/EmoRoBERTa", return_all_scores=True
+#)
 
 # Emojis
 emotions_emoji_dict = {"anger":"😠","disgust":"🤮", "fear":"😨😱", "joy":"🤗", "neutral":"😐", "sadness":"😔", "shame":"😳", "surprise":"😮"}
@@ -62,19 +62,19 @@ def main():
         
         st.subheader("Predictions from EmoRoBERTa - BERT based pre-trained model (28 emotions)")
         
-        col3,col4 = st.columns(2)
-        emotion_dataframe_emoroberta, predicted_emotion_emoroberta, prediction_probability_emoroberta = predict_emotions_and_score_emoroberta(raw_text)
-        with col3:
-            st.success("Emotion")    
-            st.write(predicted_emotion_emoroberta) 
-
-        with col4:
-            st.success("Emotion Score") 
-            st.write("{:.4f}".format(np.max(prediction_probability_emoroberta)))
-        st.success("Prediction Probability")
-        emotion_dataframe_emoroberta.columns = ["emotions","probability"]  
-        fig = alt.Chart(emotion_dataframe_emoroberta).mark_bar().encode(y=alt.Y('emotions', sort='-x'),x='probability',color='emotions')
-        st.altair_chart(fig,use_container_width=True)
+#       col3,col4 = st.columns(2)
+#       emotion_dataframe_emoroberta, predicted_emotion_emoroberta, prediction_probability_emoroberta = predict_emotions_and_score_emoroberta(raw_text)
+#       with col3:
+#           st.success("Emotion")    
+#           st.write(predicted_emotion_emoroberta) 
+#
+#        with col4:
+#            st.success("Emotion Score") 
+#            st.write("{:.4f}".format(np.max(prediction_probability_emoroberta)))
+#        st.success("Prediction Probability")
+#        emotion_dataframe_emoroberta.columns = ["emotions","probability"]  
+#        fig = alt.Chart(emotion_dataframe_emoroberta).mark_bar().encode(y=alt.Y('emotions', sort='-x'),x='probability',color='emotions')
+#        st.altair_chart(fig,use_container_width=True)
 
 if __name__ == '__main__':
     main()