Update app.py

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-st.set_page_config(page_title="ИТМО Магистратура Чат-бот", page_icon="🎓")
-st.title("🎓 Чат-бот про магистратуру ИТМО на русском")
+st.set_page_config(page_title="ИТМО Магистратура Чат-бот (Saiga Llama3 8B)", page_icon="🎓")
+st.title("🎓 Чат-бот про магистратуру ИТМО на русском с Saiga Llama3 8B")
 
-MODEL_NAME = "Grossmend/rudialogpt3_medium_based_on_gpt2"
+MODEL_NAME = "IlyaGusev/saiga_llama3_8b"
 
 @st.cache_resource
 def load_model():
@@ -20,9 +20,7 @@ tokenizer, model = load_model()
 if "history" not in st.session_state:
     st.session_state.history = []
 
-SYSTEM_PROMPT = """Вы являетесь виртуальным помощником для абитуриентов магистратуры Университета ИТМО.
-Отвечаете на вопросы о магистерских программах ИТМО.
-"""
+SYSTEM_PROMPT = """Вы являетесь виртуальным помощником для абитуриентов магистратуры Университета ИТМО. Отвечаете на вопросы о магистерских программах ИТМО."""
 
 user_input = st.text_input("Введите ваш вопрос про магистратуру ИТМО:")