Kyan14 commited on
Commit
0a7ed26
·
1 Parent(s): 53d690d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -31,9 +31,11 @@ def get_mood_from_image(image: Image.Image):
31
  mood_scores = {}
32
  for mood, score in zip(moods, probs[0]):
33
  mood_scores[mood] = score
 
 
34
 
35
  # Filter moods with a score above 60%
36
- filtered_moods = {k: v for k, v in mood_scores.items() if v > 0.6}
37
 
38
  if len(filtered_moods) < 2:
39
  return None
 
31
  mood_scores = {}
32
  for mood, score in zip(moods, probs[0]):
33
  mood_scores[mood] = score
34
+ print("Mood Scores:", mood_scores)
35
+
36
 
37
  # Filter moods with a score above 60%
38
+ filtered_moods = {k: v for k, v in mood_scores.items() if v > 0.4}
39
 
40
  if len(filtered_moods) < 2:
41
  return None