update: responses

Modarb-Ai-Trainer · Jun 22, 2024 · 6095567 · 6095567
1 parent 9c820fb
commit 6095567
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/server.py b/server.py
@@ -40,7 +40,8 @@ def fitness_predict():
             return jsonify({"error": f"{paramName} is missing"}), 400
         params[paramName] = value
 
-    return jsonify({"result": fitness_model.predict(**params)})
+    res = fitness_model.predict(**params)
+    return jsonify(res)
 
 
 @app.post("/nutrition")
@@ -53,7 +54,8 @@ def nutrition_predict():
         if value is None:
             return jsonify({"error": f"{paramName} is missing"}), 400
         params[paramName] = value
-    return jsonify({"result": nutrition_model.generate_plan(**params)})
+
+    return jsonify(nutrition_model.generate_plan(**params))
 
 
 if __name__ == "__main__":