prouditalian commited on
Commit
467b6f2
·
1 Parent(s): f6b77d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -12
app.py CHANGED
@@ -1,22 +1,11 @@
1
  import streamlit as st
2
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
- # from transformers import pipeline
4
 
5
  model_name = "ai-forever/mGPT"
6
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
7
  model = GPT2LMHeadModel.from_pretrained(model_name)
8
 
9
-
10
- # Декоратор @st.cache говорит Streamlit, что модель нужно загрузить только один раз, чтобы избежать утечек памяти
11
- # @st.cache_resource
12
- # загружает модель
13
- # def load_model():
14
- # return pipeline("text-generation", model=model, tokenizer=tokenizer)
15
-
16
-
17
- # Загружаем предварительно обученную модель
18
- # answer = load_model()
19
-
20
  # Выводим заголовок страницы
21
  st.title("Помощник студента")
22
  st.write("Приложение поможет найти ответы на ваши вопросы")
 
1
  import streamlit as st
2
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
3
+
4
 
5
  model_name = "ai-forever/mGPT"
6
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
7
  model = GPT2LMHeadModel.from_pretrained(model_name)
8
 
 
 
 
 
 
 
 
 
 
 
 
9
  # Выводим заголовок страницы
10
  st.title("Помощник студента")
11
  st.write("Приложение поможет найти ответы на ваши вопросы")