fix

app.py

@@ -15,10 +15,10 @@ model = transformers.AutoModelForSequenceClassification.from_pretrained(model_na
 if torch.cuda.is_available():
     model = model.cuda()
 
-context_3 = st.text_input('Пользователь 1', 'Привет')
-context_2 = st.text_input('Пользователь 1', 'Здарова')
-context_1 = st.text_input('Пользователь 1', 'Как жизнь?')
-response = st.text_input('Пользователь 1', 'Норм')
+context_3 = st.text_input('Настя', 'Привет')
+context_2 = st.text_input('Диалоговый агент', 'Здарова')
+context_1 = st.text_input('Настя', 'Как жизнь?')
+response = st.text_input('Диалоговый агент', 'Норм')
 sample = {
     'context_3': context_3,
     'context_2': context_2,
@@ -86,10 +86,10 @@ logits = model(**tokens).logits
 probas = torch.sigmoid(logits)[0].cpu().detach().numpy()
 
 st.metric(
-    label='Вероятность того, что последний ответ релевантный',
+    label='Вероятность того, что последний ответ диалогового агента релевантный',
     value=probas[0]
 )
 st.metric(
-    label='Вероятность того, что последний ответ специфичный',
+    label='Вероятность того, что последний ответ диалогового агента специфичный',
     value=probas[1]
 )