change graph view

app.py

@@ -33,13 +33,14 @@ if (submit and len(text.strip()) > 0) or len(text.strip()) > 0:
 
     prediction = pipe(text)[0]
     prediction = sort_predictions(prediction)
+    max_ylim = prediction[0]['score'] + 0.1
 
     fig, ax = plt.subplots()
-    ax.bar(x=[i for i, _ in enumerate(prediction)],
-           height=[p['score'] for p in prediction],
+    ax.bar(y=[i for i, _ in enumerate(prediction)],
+           width=[p['score'] for p in prediction],
            tick_label=[p['label'] for p in prediction])
     ax.tick_params(rotation=90)
-    ax.set_ylim(0, 1)
+    ax.set_ylim(0, max_ylim)
 
     st.header('Result:')
     st.pyplot(fig)