shao3d commited on
Commit
bbb2692
·
verified ·
1 Parent(s): 9d91cbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -17,7 +17,11 @@ model = PeftModel.from_pretrained(base_model, lora_repo)
17
  model.eval() # Переключаем в режим предсказания
18
 
19
  def generate_response(history):
20
- # history это список пар [пользовательское_сообщение, ответ_модели]
 
 
 
 
21
  # Берем последний вопрос пользователя
22
  user_message = history[-1][0] # Последнее сообщение пользователя
23
  # Генерируем ответ
@@ -39,7 +43,7 @@ with gr.Blocks() as demo:
39
  chatbot = gr.Chatbot()
40
  msg = gr.Textbox(placeholder="Напиши сообщение для модели...")
41
  clear = gr.Button("Очистить чат")
42
- msg.submit(generate_response, inputs=chatbot, outputs=chatbot)
43
  clear.click(lambda: [], None, chatbot)
44
 
45
  demo.launch()
 
17
  model.eval() # Переключаем в режим предсказания
18
 
19
  def generate_response(history):
20
+ # Если history пустой, берем сообщение из текстового поля (msg)
21
+ if not history: # Проверяем, пустой ли history
22
+ # Получаем последний вопрос из текстового поля через Gradio
23
+ # Но здесь мы работаем с history напрямую, поэтому предположим, что вопрос приходит через msg
24
+ raise gr.Error("Пожалуйста, введите сообщение в текстовое поле перед отправкой.") # Добавляем проверку
25
  # Берем последний вопрос пользователя
26
  user_message = history[-1][0] # Последнее сообщение пользователя
27
  # Генерируем ответ
 
43
  chatbot = gr.Chatbot()
44
  msg = gr.Textbox(placeholder="Напиши сообщение для модели...")
45
  clear = gr.Button("Очистить чат")
46
+ msg.submit(generate_response, inputs=[chatbot], outputs=chatbot) # Указываем, что входные данные — это chatbot
47
  clear.click(lambda: [], None, chatbot)
48
 
49
  demo.launch()