Update app.py

app.py

@@ -2,13 +2,15 @@ import gradio as gr
 import chromadb
 from sentence_transformers import CrossEncoder, SentenceTransformer
 import json
+from qdrant_client import QdrantClient
 
 print("Setup client")
-chroma_client = chromadb.Client()
-collection = chroma_client.create_collection(
-    name="food_collection",
-    metadata={"hnsw:space": "cosine"} # l2 is the default
-)
+#chroma_client = chromadb.Client()
+#collection = chroma_client.create_collection(
+    #name="food_collection",
+    #metadata={"hnsw:space": "cosine"} # l2 is the default
+#)
+client = QdrantClient(":memory:")
 
 print("load data")
 with open("test_json.json", "r") as f:
@@ -36,12 +38,12 @@ print('upserting')
 print("printing item:")
 embedding = embedding_function([item['doc'] for item in payload])
 print(type(embedding))
-collection.add(
+client.add(
     collection_name="food",
     documents=[item['doc'] for item in payload],
     #embeddings=embedding,
-    metadatas=[{'payload':item} for item in payload],
-    ids=[f"id_{idx}" for idx, _ in enumerate(payload)]
+    metadata=[{'payload':item} for item in payload],
+    ids=[idx for idx, _ in enumerate(payload)]
     )
 
 def search_chroma(query:str):