Update app.py

app.py

@@ -16,10 +16,12 @@ with open('tokenizer.pkl', 'rb') as tokenizer_file:
 class_mapping = {"Angry": 0, "Sad": 1, "Joy": 2, "Surprise": 3}
 numerical_to_label = {v: k for k, v in class_mapping.items()}
 
-st.title('Model Deployment')
+st.title('VibeConnect')
+
+
 
 # Text input for the user to enter a sequence
-user_input = st.text_input('Enter a sequence:')
+user_input = st.text_input('Enter a Text:')
 
 if st.button('Predict'):
     # Tokenize and pad the user input
@@ -29,8 +31,10 @@ if st.button('Predict'):
     # Make predictions
     prediction = lstm_model.predict(padded_sequence)
 
-    # Display the predicted probabilities and labels
-    st.write('Predicted Probabilities:')
+    threshold = 0.5 
+    # Display the label
     for i in range(len(prediction[0])):
         label = numerical_to_label[i]
-        st.write(f'{label}: {prediction[0][i]}')
+        probability = prediction[0][i]
+        if probability > threshold:
+            st.write(f'{label}: {probability}')