d.tsimerman commited on
Commit
ef42e41
1 Parent(s): a922c26

new models

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/crossencoder-tiny', 'tinkoff-ai/crossencoder-medium', 'tinkoff-ai/crossencoder-large')
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)