flowise / app.py
MORENdi's picture
Create app.py
7189e33 verified
raw
history blame contribute delete
935 Bytes
import streamlit as st
from transformers import pipeline # Или другой способ загрузки вашей модели
# Загрузка модели (один раз при запуске приложения)
@st.cache_resource
def load_model():
return pipeline("text-generation", model="DeepPavlov/rubert-base-cased-conversational")
# Обработка текста
def process_text_with_model(text):
# Логика для обработки текста с помощью вашей модели
# Пример:
generator = load_model()
response = generator(text, max_length=100, num_return_sequences=1)[0]['generated_text']
return response
# Интерфейс Streamlit
st.title("Чат-бот по \"Аленькому цветочку\"")
user_input = st.text_input("Введите ваш вопрос:")
if user_input:
response = process_text_with_model(user_input)
st.write(response)