Update app.py

app.py

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