Sushrut98 commited on
Commit
e2e7758
·
verified ·
1 Parent(s): ea7e208

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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 = example_list[0]
69
  a = a[0]
70
  return pred_labels_and_probs,a
71
  elif emoji == 'disgust':
72
- a = example_list[1]
73
  a = a[0]
74
  return pred_labels_and_probs,a
75
  elif emoji == 'fear':
76
- a = example_list[2]
77
  a = a[0]
78
  return pred_labels_and_probs,a
79
  elif emoji == 'happy':
80
- a = example_list[3]
81
  a = a[0]
82
  return pred_labels_and_probs,a
83
  elif emoji == 'neutral':
84
- a = example_list[4]
85
  a = a[0]
86
  return pred_labels_and_probs,a
87
  elif emoji == 'sad':
88
- a = example_list[5]
89
  a = a[0]
90
  return pred_labels_and_probs,a
91
  elif emoji == 'surprise':
92
- a = example_list[6]
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