Update main.py

main.py

@@ -69,9 +69,9 @@ class MyEmbeddingFunction(EmbeddingFunction):
                 }
 
                 response = requests.post(url, headers=headers, json=payload)
-                return response.json()["embeddings"][0]
-            except:
-                pass
+                return response.json()
+            except Exception as e:
+                print("Error in Embeding :",str(e))
 
 try:
     CHROMA_PATH = "chroma"
@@ -117,7 +117,7 @@ def strings_ranked_by_relatedness(query, df, top_n=5):
         return np.dot(x, y) / (x_norm * y_norm)
 
     query_embedding_response = embeddingGen(query)
-    query_embedding = query_embedding_response["embeddings"][0]
+    query_embedding = query_embedding_response
     strings_and_relatednesses = [
         (row["text"], relatedness_fn(query_embedding, row["embedding"])) for row in df
     ]