diff --git a/api/app.py b/api/app.py
index b3fc58f6..2006869d 100644
--- a/api/app.py
+++ b/api/app.py
@@ -22,23 +22,21 @@
def hello_world():
return jsonify("Hello, world!")
-
@app.route('/classify', methods=["POST"])
def upload_file():
received_data = request.get_json()
selected_data = received_data.get('data')
- selected_classifier = received_data.get('classifier')
+ selected_classifier = int(received_data.get('classifier'))
- print("selected data: ", selected_data)
- print("selected classifier: ", selected_classifier)
+ #print("selected data: ", selected_data)
+ #print("selected classifier: ", selected_classifier)
df = pd.DataFrame(selected_data, columns=['input_column'])
result = data_processer.handle_classify(df, selected_classifier)
return jsonify({'result': result.to_json()})
-
@app.route('/nb-news-model', methods=["POST"])
def news_model():
result = data_processer.nb_news_application()
@@ -55,7 +53,5 @@ def lin_regression_model():
result = data_processer.lin_regression_model()
return jsonify({"result": result.to_json()})
-
-
if __name__ == '__main__':
app.run(host='0.0.0.0', port=5000, debug=True)
diff --git a/src/pages/home.tsx b/src/pages/home.tsx
index 205985f2..13fac0a0 100644
--- a/src/pages/home.tsx
+++ b/src/pages/home.tsx
@@ -32,7 +32,7 @@ export default function Home() {
classifier: selectedClassifier,
})
.catch((error) => {
- console.log(error);
+ console.error(error.response.data);
});
if (response && response.data) {
@@ -123,8 +123,9 @@ export default function Home() {
-
-
+
+
+