d.tsimerman commited on
Commit
c7ca481
1 Parent(s): 2c9fa73
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import os
2
  import streamlit as st
3
  import transformers
4
  import torch
 
5
  from typing import List, Dict
6
 
7
  st.subheader('Эта демонстрация позволяет поэксперементировать с моделями, которые оценивают, насколько предлагаемый ответ подходит к контексту диалога.')
@@ -11,7 +12,7 @@ model_name = st.selectbox(
11
  )
12
  auth_token = os.environ.get('TOKEN') or True
13
 
14
- @st.cache(hash_funcs={transformers.Tokenizer: lambda tokenizer: hash(tokenizer.name_or_path)})
15
  def load_model(model_name: str):
16
  with st.spinner('Loading models...'):
17
  tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
2
  import streamlit as st
3
  import transformers
4
  import torch
5
+ import tokenizers
6
  from typing import List, Dict
7
 
8
  st.subheader('Эта демонстрация позволяет поэксперементировать с моделями, которые оценивают, насколько предлагаемый ответ подходит к контексту диалога.')
12
  )
13
  auth_token = os.environ.get('TOKEN') or True
14
 
15
+ @st.cache(hash_funcs={tokenizers.Tokenizer: lambda tokenizer: hash(tokenizer.name_or_path)})
16
  def load_model(model_name: str):
17
  with st.spinner('Loading models...'):
18
  tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)