File size: 935 Bytes
7189e33
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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)