Spaces:
Sleeping
Sleeping
File size: 1,247 Bytes
188456f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import streamlit as st
from joblib import load
from sklearn.feature_extraction.text import TfidfVectorizer
# Загрузка сохраненной модели
clf = load('my_model_filename.joblib')
# Можно также загрузить и другие инструменты, если они были сохранены (например, TF-IDF векторизатор)
# vectorizer = load('vectorizer_filename.joblib')
# Запуск приложения
def main():
st.title("Модель классификации отзывов")
# Ввод текста
user_input = st.text_area("Введите текст отзыва:")
if st.button("Классифицировать"):
# Векторизация текста (если вы использовали TF-IDF или другой векторизатор)
# user_input_vec = vectorizer.transform([user_input])
# Прогноз модели
prediction = clf.predict(user_input) # Используйте user_input_vec вместо user_input, если текст нужно векторизировать
st.write(f"Прогнозируемый класс: {prediction[0]}")
if __name__ == "__main__":
main() |