Update app.py

app.py

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