Spaces:
Sleeping
Sleeping
Commit ·
41b4734
1
Parent(s): 0d88129
Update functions.py
Browse files- functions.py +7 -4
functions.py
CHANGED
|
@@ -40,10 +40,13 @@ def prediction(X):
|
|
| 40 |
model = load_model('best_model.keras')
|
| 41 |
y_pred = model.predict(X)
|
| 42 |
predictions = np.argmax(y_pred, axis=1)
|
| 43 |
-
for
|
| 44 |
if val == 0:
|
| 45 |
-
|
|
|
|
| 46 |
elif val == 1:
|
| 47 |
-
|
|
|
|
| 48 |
else:
|
| 49 |
-
|
|
|
|
|
|
| 40 |
model = load_model('best_model.keras')
|
| 41 |
y_pred = model.predict(X)
|
| 42 |
predictions = np.argmax(y_pred, axis=1)
|
| 43 |
+
for val in predictions:
|
| 44 |
if val == 0:
|
| 45 |
+
emo=0
|
| 46 |
+
return f"You are feeling Fear"
|
| 47 |
elif val == 1:
|
| 48 |
+
emo=1
|
| 49 |
+
return f"You are feeling Anger"
|
| 50 |
else:
|
| 51 |
+
emo=2
|
| 52 |
+
return f"You are feeling Joy"
|