Update main.py

main.py

@@ -39,7 +39,7 @@ class MyEmbeddingFunction(EmbeddingFunction):
                 }
 
                 response = requests.post(url, headers=headers, json=payload)
-                return response.json()["embeddings"]
+                return response.json()[0][0]
             except:
                 pass
 
@@ -69,7 +69,7 @@ class MyEmbeddingFunction(EmbeddingFunction):
                 }
 
                 response = requests.post(url, headers=headers, json=payload)
-                return response.json()
+                return response.json()[0][0]
             except Exception as e:
                 print("Error in Embeding :",str(e))
 
@@ -107,7 +107,7 @@ def embeddingGen(query):
     }
 
     response = requests.post(url, headers=headers, json=payload)
-    return response.json()
+    return response.json()[0][0]
 
 
 def strings_ranked_by_relatedness(query, df, top_n=5):