Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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
|