category list derived

app.py

@@ -43,7 +43,8 @@ def main():
         model = load_learner(MODEL_PATH)
         model_output = model.predict('fruit_image.jpg')
 
-        prob_idx = model.dls.vocab.index(model_output[0])
+        category_list = [cat for cat in model.dls.vocab]
+        prob_idx = category_list.index(model_output[0])
         st.write(f'{model_output[0].title()} is depicted in the photo with {model_output[-1][prob_idx]:.4f} confidence.')
 
         st.session_state.pop("fruit")