Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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%).")
|