Update app.py

app.py

@@ -95,7 +95,11 @@ def main():
             lstm_model = load_lstm_model(model_path)
             # Predict hate speech
             prediction = predict_hate_speech(input_text, lstm_model)
-            st.success("result : " + str(prediction))
+            # Convert the list to a numpy array
+            arr = np.array(prediction[0])
+            max_index = np.argmax(arr)
+
+            st.success("class"+str(max_index))
             #if prediction > 0.5:
             #    st.error("Hate Speech Detected")
             #else: