Upload folder using huggingface_hub

app.py

@@ -32,7 +32,7 @@ def predict():
     }
     input_data = pd.DataFrame([sample])
     prediction = model.predict(input_data).tolist()[0]
-    return jsonify({"Predicted": prediction})
+    return jsonify({"Prediction": prediction})
 
 
 @app.post("/v1/productbatch")
@@ -40,7 +40,7 @@ def predict_batch():
     file = request.files["file"]
     input_data = pd.read_csv(file)
     predictions = model.predict(input_data).tolist()
-    input_data["Predicted"] = predictions
+    input_data["Prediction"] = predictions
     result = input_data.to_dict(orient="records")
     return jsonify(result)