Update app.py

app.py

@@ -58,7 +58,7 @@ def get_all_result(prediction):
     dict = {}
     for element in prediction:
         for i in range(0, len(element)):
-            dict[element[i]] = le_departure.inverse_transform([i])
+            dict[element[i]] = le_departure.inverse_transform([i])[0]
     return dict
 
 
@@ -102,7 +102,7 @@ def main():
                 st.write("{}:{}".format(res, proba))
                 st.write("Confidence:{}".format(proba))
 
-                source = pd.DataFrame({'Proba': [1,2,3], 'Emotion': ['a','b','c']})
+                source = pd.DataFrame({'Proba': list(total_result.keys()), 'Emotion': list(total_result.values())})
                 st.write(source)
 
                 fig = alt.Chart(source).mark_bar().encode(x='Emotion',y='Proba',color='Emotion')