Yerzhxn commited on
Commit
f5d4c5a
·
verified ·
1 Parent(s): 928bca7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ if st.button("Предсказать"):
70
  # Проверка вероятности для отображения результата
71
  if max_prob.item() > 0.35:
72
  st.write(f"Предсказанный класс: {predicted_class.item()}, вероятность: {max_prob.item():.2f}")
73
- st.write(df[df['label']==predicted_class.item()])
74
  else:
75
  st.write("Модель не уверена в предсказании (вероятность меньше 35%).")
76
  else:
 
70
  # Проверка вероятности для отображения результата
71
  if max_prob.item() > 0.35:
72
  st.write(f"Предсказанный класс: {predicted_class.item()}, вероятность: {max_prob.item():.2f}")
73
+ st.write(df[df['label']==predicted_class.item()]['PROF_NAME'])
74
  else:
75
  st.write("Модель не уверена в предсказании (вероятность меньше 35%).")
76
  else: