Update app.py

app.py

@@ -14,7 +14,8 @@ model = model.to(device)
 
 def respond(message, history):
     history = history or []
-    # Объединяем историю и текущий ввод
+
+    # Собираем всю историю в виде текста
     full_input = ""
     for user, bot in history:
         full_input += f"User: {user}\nAssistant: {bot}\n"
@@ -29,16 +30,18 @@ def respond(message, history):
         top_p=0.95,
         pad_token_id=tokenizer.eos_token_id
     )
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    
-    # Извлекаем только ответ ассистента (после последнего "Assistant:")
-    if "Assistant:" in response:
-        response = response.split("Assistant:")[-1].strip()
-    
+    output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    # Извлекаем только ответ ассистента
+    response = output_text.split("Assistant:")[-1].strip()
+
+    # Добавляем в историю в виде (вопрос, ответ)
     history.append((message, response))
+
     return response, history
 
+
 chat = gr.ChatInterface(fn=respond, title="Innopolis Chatbot")
 
 if __name__ == "__main__":
-    chat.launch()
+    chat.launch(share=True)