modified to make knn flexible

app.py

@@ -15,15 +15,18 @@ app = Flask(__name__)
 def query():
     """
     {
-        embeddings: <embeddings_list>
+        embeddings: <embeddings_list>,
+        knn: 3
     }
     """
 
-    embeddings = request.get_json()['embeddings']
+    data = request.get_json()
+    embeddings = data['embeddings']
+    top_k = data['knn']
 
     matches = index.query(
         vector = embeddings,
-        top_k = 3,
+        top_k = top_k,
         include_metadata = True
     )