oiisa commited on
Commit
7c9d529
·
verified ·
1 Parent(s): fdfb963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,10 +2,11 @@ import streamlit as st
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
- st.set_page_config(page_title="ИТМО Магистратура Чат-бот (Saiga Llama3 8B)", page_icon="🎓")
6
- st.title("🎓 Чат-бот про магистратуру ИТМО на русском с Saiga Llama3 8B")
 
 
7
 
8
- MODEL_NAME = "IlyaGusev/saiga_llama3_8b"
9
 
10
  @st.cache_resource
11
  def load_model():
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
+ st.set_page_config(page_title="ИТМО Магистратура Чат-бот", page_icon="🎓")
6
+ st.title("🎓 Чат-бот про магистратуру ИТМО")
7
+
8
+ MODEL_NAME = "Grossmend/rudialogpt3_medium_based_on_gpt2"
9
 
 
10
 
11
  @st.cache_resource
12
  def load_model():