Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def show_likelihood(pred_label):
|
|
28 |
class_probs = pred_label[2].numpy()
|
29 |
classes = ["Temple", "You", "Me", "You", "Friend", "Love", "Enough", "Church","Mosque"]
|
30 |
class_labels = [classes[i] for i in range(len(class_probs))]
|
31 |
-
fig =
|
32 |
pt.barh(class_labels, class_probs)
|
33 |
pt.ylabel("Class")
|
34 |
pt.xlabel("Probability")
|
|
|
28 |
class_probs = pred_label[2].numpy()
|
29 |
classes = ["Temple", "You", "Me", "You", "Friend", "Love", "Enough", "Church","Mosque"]
|
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")
|