Spaces:
Sleeping
Sleeping
asrvastava
commited on
Commit
·
cd4f98e
1
Parent(s):
34c147f
edit
Browse files- __pycache__/main.cpython-39.pyc +0 -0
- main.py +3 -3
__pycache__/main.cpython-39.pyc
CHANGED
Binary files a/__pycache__/main.cpython-39.pyc and b/__pycache__/main.cpython-39.pyc differ
|
|
main.py
CHANGED
@@ -36,12 +36,12 @@ def predict(data: PredictRequest):
|
|
36 |
features = [data.AG_X, data.AG_Y, data.AG_Z, data.Acc_X, data.Acc_Y, data.Acc_Z, data.Gravity_X, data.Gravity_Y,
|
37 |
data.Gravity_Z, data.RR_X, data.RR_Y, data.RR_Z, data.RV_X, data.RV_Y, data.RV_Z, data.cos]
|
38 |
|
39 |
-
predictions = loaded_model.predict([features])
|
40 |
-
|
41 |
|
42 |
activities = ["Running", "Sitting", "Standing", "Walking", "DownStairs", "Upstairs"]
|
43 |
activity = activities[predictions]
|
44 |
|
45 |
-
return {'activity':
|
46 |
|
47 |
|
|
|
36 |
features = [data.AG_X, data.AG_Y, data.AG_Z, data.Acc_X, data.Acc_Y, data.Acc_Z, data.Gravity_X, data.Gravity_Y,
|
37 |
data.Gravity_Z, data.RR_X, data.RR_Y, data.RR_Z, data.RV_X, data.RV_Y, data.RV_Z, data.cos]
|
38 |
|
39 |
+
predictions = loaded_model.predict([features])[0]
|
40 |
+
print(predictions)
|
41 |
|
42 |
activities = ["Running", "Sitting", "Standing", "Walking", "DownStairs", "Upstairs"]
|
43 |
activity = activities[predictions]
|
44 |
|
45 |
+
return {'activity': activity}
|
46 |
|
47 |
|