Spaces:
Runtime error
Runtime error
d.tsimerman
commited on
Commit
•
c7ca481
1
Parent(s):
2c9fa73
good
Browse files
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={
|
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)
|