drsaikirant88 commited on
Commit
94858c4
1 Parent(s): 78c42e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -255,7 +255,7 @@ def predict(img):
255
  emotions = {learn_emotion_labels[i]: float(probs_emotion[i]) for i in range(len(learn_emotion_labels))}
256
  sentiments = {learn_sentiment_labels[i]: float(probs_sentiment[i]) for i in range(len(learn_sentiment_labels))}
257
 
258
- output = [img, emotions, sentiments, None, None, None, None, None, None]
259
 
260
  else: # Max 3 for now
261
  for face in faces[:3]:
 
255
  emotions = {learn_emotion_labels[i]: float(probs_emotion[i]) for i in range(len(learn_emotion_labels))}
256
  sentiments = {learn_sentiment_labels[i]: float(probs_sentiment[i]) for i in range(len(learn_sentiment_labels))}
257
 
258
+ output = [img, emotions, sentiments, img, emotions, sentiments, img, emotions, sentiments]
259
 
260
  else: # Max 3 for now
261
  for face in faces[:3]: