Spaces:

KaburaJ
/

KSL_imageClassificationApp

Runtime error

KaburaJ commited on Feb 12, 2023

Commit

ed0631e

•

1 Parent(s): bb0e5b8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def predict(image_path):
 def show_likelihood(pred_label):
     class_probs = pred_label[2].numpy()
-    classes = ["Temple", "You", "Me", "You", "Friend", "Love", "Enough", "Church","Mosque"]
     class_labels = [classes[i] for i in range(len(class_probs))]
     fig = pt.figure(figsize=(10, 10))
     pt.barh(class_labels, class_probs)

 def show_likelihood(pred_label):
     class_probs = pred_label[2].numpy()
+    classes = ["Temple", "You", "Me", "Seat", "Friend", "Love", "Enough", "Church","Mosque"]
     class_labels = [classes[i] for i in range(len(class_probs))]
     fig = pt.figure(figsize=(10, 10))
     pt.barh(class_labels, class_probs)