Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,18 +24,18 @@ def predict(image_path):
|
|
24 |
return pred_label
|
25 |
|
26 |
|
27 |
-
def show_likelihood(pred_label):
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
|
40 |
def main():
|
41 |
|
@@ -62,7 +62,7 @@ def main():
|
|
62 |
pred_label = predict(image_path)
|
63 |
st.write("The image was classified as:", pred_label[0])
|
64 |
|
65 |
-
show_likelihood(pred_label)
|
66 |
|
67 |
if __name__ == '__main__':
|
68 |
main()
|
|
|
24 |
return pred_label
|
25 |
|
26 |
|
27 |
+
# def show_likelihood(pred_label):
|
28 |
+
# class_probs = pred_label[9].numpy()
|
29 |
+
# classes = pred_label[9]
|
30 |
+
# class_labels = [classes[i] for i in range(len(class_probs))]
|
31 |
+
# fig = pt.figure(figsize=(10, 10))
|
32 |
+
# pt.barh(class_labels, class_probs)
|
33 |
+
# pt.ylabel("Class")
|
34 |
+
# pt.xlabel("Probability")
|
35 |
+
# pt.title("Class Probabilities")
|
36 |
+
# pt.xlim(0, 1)
|
37 |
+
# pt.ylim(-1, len(class_probs))
|
38 |
+
# st.pyplot(fig)
|
39 |
|
40 |
def main():
|
41 |
|
|
|
62 |
pred_label = predict(image_path)
|
63 |
st.write("The image was classified as:", pred_label[0])
|
64 |
|
65 |
+
# show_likelihood(pred_label)
|
66 |
|
67 |
if __name__ == '__main__':
|
68 |
main()
|