Update app.py

app.py

@@ -24,18 +24,18 @@ def predict(image_path):
     return pred_label
 
 
-def show_likelihood(pred_label):
-    class_probs = pred_label[9].numpy()
-    classes = pred_label[9]
-    class_labels = [classes[i] for i in range(len(class_probs))]
-    fig = pt.figure(figsize=(10, 10))
-    pt.barh(class_labels, class_probs)
-    pt.ylabel("Class")
-    pt.xlabel("Probability")
-    pt.title("Class Probabilities")
-    pt.xlim(0, 1)
-    pt.ylim(-1, len(class_probs))
-    st.pyplot(fig)
+# def show_likelihood(pred_label):
+#     class_probs = pred_label[9].numpy()
+#     classes = pred_label[9]
+#     class_labels = [classes[i] for i in range(len(class_probs))]
+#     fig = pt.figure(figsize=(10, 10))
+#     pt.barh(class_labels, class_probs)
+#     pt.ylabel("Class")
+#     pt.xlabel("Probability")
+#     pt.title("Class Probabilities")
+#     pt.xlim(0, 1)
+#     pt.ylim(-1, len(class_probs))
+#     st.pyplot(fig)
 
 def main():
     
@@ -62,7 +62,7 @@ def main():
         pred_label = predict(image_path)
         st.write("The image was classified as:", pred_label[0])
 
-        show_likelihood(pred_label)
+        # show_likelihood(pred_label)
 
 if __name__ == '__main__':
     main()