Update app.py

app.py

@@ -24,18 +24,22 @@ languages = [
 selected_language = st.selectbox('Select the language:', languages)
 user_text = st.text_input('Enter the text:')
 
-encoder = LaserEncoderPipeline(lang=language)
+encoder = LaserEncoderPipeline(lang=selected_language)  # Fix the variable name
+
+target_classes = [1, 2, 3]
+label_encoder.fit(target_classes)
 
 user_text_embedding = encoder.encode_sentences([user_text])[0]
 user_text_embedding = np.reshape(user_text_embedding, (1, -1))
 
 predicted_sentiment = np.argmax(model.predict(user_text_embedding))
 predicted_sentiment_no = label_encoder.inverse_transform([predicted_sentiment])[0]
+
 if predicted_sentiment_no == 1:
-  predicted_sentiment_label = 'neutral'
+    predicted_sentiment_label = 'neutral'
 elif predicted_sentiment_no == 2:
-  predicted_sentiment_label = 'positive'
+    predicted_sentiment_label = 'positive'
 else:
-  predicted_sentiment_label = 'negative'
+    predicted_sentiment_label = 'negative'
 
-st.write("Predicted Sentiment:" + predicted_sentiment_label)
+st.write("Predicted Sentiment:" + predicted_sentiment_label)