Spaces:
Runtime error
Runtime error
d.tsimerman
commited on
Commit
•
ef42e41
1
Parent(s):
a922c26
new models
Browse files
app.py
CHANGED
@@ -4,10 +4,10 @@ import transformers
|
|
4 |
import torch
|
5 |
from typing import List, Dict
|
6 |
|
7 |
-
st.subheader('Эта демонстрация позволяет поэксперементировать с моделями, которые оценивают, насколько подходит предлагаемый ответ к контексту
|
8 |
model_name = st.selectbox(
|
9 |
'Выберите модель',
|
10 |
-
('tinkoff-ai/
|
11 |
)
|
12 |
auth_token = os.environ.get('TOKEN') or True
|
13 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
|
|
|
4 |
import torch
|
5 |
from typing import List, Dict
|
6 |
|
7 |
+
st.subheader('Эта демонстрация позволяет поэксперементировать с моделями, которые оценивают, насколько подходит предлагаемый ответ к контексту диалога.')
|
8 |
model_name = st.selectbox(
|
9 |
'Выберите модель',
|
10 |
+
('tinkoff-ai/response-quality-classifier-tiny', 'tinkoff-ai/response-quality-classifier-base', 'tinkoff-ai/response-quality-classifier-large')
|
11 |
)
|
12 |
auth_token = os.environ.get('TOKEN') or True
|
13 |
tokenizer = transformers.AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
|