Update app.py

app.py

@@ -16,7 +16,7 @@ def predict(input_data):
     input_data = np.array(input_data).reshape(1, -1)
     # Make predictions
     predictions = knn.predict([[0.2,0.03,0.0,1.0,0.0]])
-    return input_data
+    return predictions[0]