Spaces:
Sleeping
Sleeping
asrvastava
commited on
Commit
·
be46641
1
Parent(s):
cd4f98e
label changes
Browse files
main.py
CHANGED
|
@@ -39,7 +39,7 @@ def predict(data: PredictRequest):
|
|
| 39 |
predictions = loaded_model.predict([features])[0]
|
| 40 |
print(predictions)
|
| 41 |
|
| 42 |
-
activities = ["
|
| 43 |
activity = activities[predictions]
|
| 44 |
|
| 45 |
return {'activity': activity}
|
|
|
|
| 39 |
predictions = loaded_model.predict([features])[0]
|
| 40 |
print(predictions)
|
| 41 |
|
| 42 |
+
activities = ["Walking", "Sitting", "Standing", "Sitting", "DownStairs", "Upstairs"]
|
| 43 |
activity = activities[predictions]
|
| 44 |
|
| 45 |
return {'activity': activity}
|