Update main.py

main.py

@@ -2,11 +2,9 @@ import streamlit as st
 from joblib import load
 from sklearn.feature_extraction.text import TfidfVectorizer
 
-# Загрузка сохраненной модели
-clf = load('my_model_filename.joblib')
+clf = load('my_model_filename.pkl')
+vectorizer = TfidfVectorizer()
 
-# Можно также загрузить и другие инструменты, если они были сохранены (например, TF-IDF векторизатор)
-# vectorizer = load('vectorizer_filename.joblib')
 
 # Запуск приложения
 def main():
@@ -17,9 +15,10 @@ def main():
     
     if st.button("Классифицировать"):
         # Векторизация текста (если вы использовали TF-IDF или другой векторизатор)
-        # user_input_vec = vectorizer.transform([user_input])
+        user_input_vec = vectorizer.transform([user_input])
+        sentence_vector_scaled = scaler.transform(sentence_vector)
         # Прогноз модели
-        prediction = clf.predict(user_input)  # Используйте user_input_vec вместо user_input, если текст нужно векторизировать
+        prediction = clf.predict(sentence_vector_scaled)  # Используйте user_input_vec вместо user_input, если текст нужно векторизировать
         st.write(f"Прогнозируемый класс: {prediction[0]}")
 
 if __name__ == "__main__":