Yerzhxn commited on
Commit
fc22fb7
·
verified ·
1 Parent(s): faf3ebe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ if st.button("Предсказать"):
66
  max_prob, predicted_class = torch.max(probabilities, dim=1)
67
 
68
  # Проверка вероятности для отображения результата
69
- if max_prob.item() > 0.5:
70
  st.write(f"Предсказанный класс: {predicted_class.item()}, вероятность: {max_prob.item():.2f}")
71
  else:
72
  st.write("Модель не уверена в предсказании (вероятность меньше 50%).")
 
66
  max_prob, predicted_class = torch.max(probabilities, dim=1)
67
 
68
  # Проверка вероятности для отображения результата
69
+ if max_prob.item() > 0.35:
70
  st.write(f"Предсказанный класс: {predicted_class.item()}, вероятность: {max_prob.item():.2f}")
71
  else:
72
  st.write("Модель не уверена в предсказании (вероятность меньше 50%).")