vortex123 commited on
Commit
3268104
1 Parent(s): 5c72962

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -18,11 +18,14 @@ with gr.Blocks() as demo:
18
 
19
  # Функция для обработки сообщений пользователя и получения ответов от модели
20
  def chat(message):
21
- # Формирование входных данных для модели
22
- chat_history = [(message, "user")] # Пользовательское сообщение
23
- # Вызов модели для генерации ответа
24
- response = interface([system_prompt.value, message]) # Пример вызова модели, зависит от реализации
25
- chat_history.append((response, "assistant")) # Ответ ассистента
 
 
 
26
  return chat_history
27
 
28
  msg = gr.Textbox(label="Type your message here...", placeholder="Enter your message...")
 
18
 
19
  # Функция для обработки сообщений пользователя и получения ответов от модели
20
  def chat(message):
21
+ # Формируем входные данные для модели включая системный промпт и сообщение пользователя
22
+ inputs = [system_prompt.value.strip(), message.strip()]
23
+ # Вызов модели для генерации ответа с увеличением лимита токенов до 100
24
+ response = interface(inputs, parameters={"max_tokens": 100})
25
+ # Ожидаем, что ответ будет в формате списка строк, каждая строка - это сообщение
26
+ chat_history = []
27
+ for part in response:
28
+ chat_history.append((part, "user" if part == message else "assistant"))
29
  return chat_history
30
 
31
  msg = gr.Textbox(label="Type your message here...", placeholder="Enter your message...")