Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from transformers import pipeline
 import torch
 import matplotlib.pyplot as plt
+import nump as np
 #from PIL import Image
 
 
@@ -80,7 +81,7 @@ if st.button('Submit'):
         ax.set_ylabel("Predicted category")
         st.pyplot(fig)
         
-        st.write('Category: {} | Probability: {:.1f}%'.format(labels_plot[number],(probs_plot[predictions])*1))
+        st.write('Category: {} | Probability: {:.1f}%'.format(labels_plot[np.array(number)],(probs_plot[predictions])*1))
    #     output = genQuestion(option, input)
    #     print(output)
    # st.write(output)