Spaces:
Sleeping
Sleeping
import streamlit as st | |
from joblib import load | |
clf = load('my_model_filename.pkl') | |
vectorizer = load('tfidf_vectorizer.pkl') | |
scaler = load('scaler.joblib') | |
# Запуск приложения | |
def main(): | |
st.title("Модель классификации отзывов") | |
# Ввод текста | |
user_input = st.text_area("Введите текст отзыва:") | |
if st.button("Классифицировать"): | |
# Векторизация текста (если вы использовали TF-IDF или другой векторизатор) | |
user_input_vec = vectorizer.transform([user_input]) | |
sentence_vector_scaled = scaler.transform(user_input_vec) | |
# Прогноз модели | |
prediction = clf.predict( | |
sentence_vector_scaled) # Используйте user_input_vec вместо user_input, если текст нужно векторизировать | |
st.write(f"Прогнозируемый класс: {prediction[0]}") | |
if __name__ == "__main__": | |
main() |