gfix

app.py

@@ -31,8 +31,8 @@ with st.spinner("Loading model..."):
 
 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
+    sorted_p = sort_predictions(prediction)
+    max_ylim = sorted_p[0]['score'] + 0.1
     fig, ax = plt.subplots()
     ax.barh([p['label'] for p in prediction], [p['score'] for p in prediction])
     #ax.tick_params(rotation=0)