Spaces:
Sleeping
Sleeping
Commit ·
17bc1b4
1
Parent(s): 0c475de
Update functions.py
Browse files- functions.py +3 -6
functions.py
CHANGED
|
@@ -42,11 +42,8 @@ def prediction(X):
|
|
| 42 |
predictions = np.argmax(y_pred, axis=1)
|
| 43 |
for val in predictions:
|
| 44 |
if val == 0:
|
| 45 |
-
|
| 46 |
-
return f"You are feeling Fear"
|
| 47 |
elif val == 1:
|
| 48 |
-
|
| 49 |
-
return f"You are feeling Anger"
|
| 50 |
else:
|
| 51 |
-
|
| 52 |
-
return f"You are feeling Joy"
|
|
|
|
| 42 |
predictions = np.argmax(y_pred, axis=1)
|
| 43 |
for val in predictions:
|
| 44 |
if val == 0:
|
| 45 |
+
return 0, f"You are feeling Fear"
|
|
|
|
| 46 |
elif val == 1:
|
| 47 |
+
return 1, f"You are feeling Anger"
|
|
|
|
| 48 |
else:
|
| 49 |
+
return 2, f"You are feeling Joy"
|
|
|