Update app.py
Browse files
app.py
CHANGED
@@ -65,31 +65,31 @@ def predict(img) :
|
|
65 |
emoji_1 = torch.argmax(pred_probs)
|
66 |
emoji = class_names[emoji_1]
|
67 |
if emoji == 'angry':
|
68 |
-
a =
|
69 |
a = a[0]
|
70 |
return pred_labels_and_probs,a
|
71 |
elif emoji == 'disgust':
|
72 |
-
a =
|
73 |
a = a[0]
|
74 |
return pred_labels_and_probs,a
|
75 |
elif emoji == 'fear':
|
76 |
-
a =
|
77 |
a = a[0]
|
78 |
return pred_labels_and_probs,a
|
79 |
elif emoji == 'happy':
|
80 |
-
a =
|
81 |
a = a[0]
|
82 |
return pred_labels_and_probs,a
|
83 |
elif emoji == 'neutral':
|
84 |
-
a =
|
85 |
a = a[0]
|
86 |
return pred_labels_and_probs,a
|
87 |
elif emoji == 'sad':
|
88 |
-
a =
|
89 |
a = a[0]
|
90 |
return pred_labels_and_probs,a
|
91 |
elif emoji == 'surprise':
|
92 |
-
a =
|
93 |
a = a[0]
|
94 |
return pred_labels_and_probs,a
|
95 |
|
|
|
65 |
emoji_1 = torch.argmax(pred_probs)
|
66 |
emoji = class_names[emoji_1]
|
67 |
if emoji == 'angry':
|
68 |
+
a = emoji_list[0]
|
69 |
a = a[0]
|
70 |
return pred_labels_and_probs,a
|
71 |
elif emoji == 'disgust':
|
72 |
+
a = emoji_list[1]
|
73 |
a = a[0]
|
74 |
return pred_labels_and_probs,a
|
75 |
elif emoji == 'fear':
|
76 |
+
a = emoji_list[2]
|
77 |
a = a[0]
|
78 |
return pred_labels_and_probs,a
|
79 |
elif emoji == 'happy':
|
80 |
+
a = emoji_list[3]
|
81 |
a = a[0]
|
82 |
return pred_labels_and_probs,a
|
83 |
elif emoji == 'neutral':
|
84 |
+
a = emoji_list[4]
|
85 |
a = a[0]
|
86 |
return pred_labels_and_probs,a
|
87 |
elif emoji == 'sad':
|
88 |
+
a = emoji_list[5]
|
89 |
a = a[0]
|
90 |
return pred_labels_and_probs,a
|
91 |
elif emoji == 'surprise':
|
92 |
+
a = emoji_list[6]
|
93 |
a = a[0]
|
94 |
return pred_labels_and_probs,a
|
95 |
|