Update app.py

app.py

@@ -18,10 +18,10 @@ if st.button("Calculate Scores"):
     scores = inference(user_input)
     st.write(scores)
     st.write(categories)
-    #scores = [round(score * 2) / 2 for score in scores]
-    #new_table = {categories[i]: scores[i] for i in range(len(categories))}
-    #scores_df = pd.DataFrame(new_table, index=['Score'])
-    #st.table(scores_df)
+    scores = [round(score * 2) / 2 for score in scores[0]]
+    new_table = {categories[i]: scores[0][i] for i in range(len(categories))}
+    scores_df = pd.DataFrame(new_table, index=['Score'])
+    st.table(scores_df)
 
 # Display the initial scores table
 st.table(scores_df)