Update app.py

app.py

@@ -70,7 +70,9 @@ if st.button("Предсказать"):
     # Проверка вероятности для отображения результата
     if max_prob.item() > 0.35:
         st.write(f"Предсказанный класс: {predicted_class.item()}, вероятность: {max_prob.item():.2f}")
-        st.write(df[df['label']==predicted_class.item()]['PROF_NAME'])
+        dataframe = df[df['label']==predicted_class.item()]
+        str1 = dataframe['PROF_NAME'][0]
+        st.write(str1)
     else:
         st.write("Модель не уверена в предсказании (вероятность меньше 35%).")
 else: