triying embeddings

Dockerfile

@@ -1,7 +1,7 @@
 # Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.9
+FROM python:3.12.3
 
 RUN useradd -m -u 1000 user
 USER user

app.py

@@ -1,9 +1,21 @@
 from fastapi import FastAPI
 from os import getenv
+from langchain_huggingface import HuggingFaceEmbeddings
 
 app = FastAPI()
 MY_KEY = getenv("MY_KEY")
 
+embeddings = HuggingFaceEmbeddings(model_name="jinaai/jina-embeddings-v2-small-en")
+
+
+@app.get("/embeddings")
+def get_embeddings():
+    test = embeddings.embed_query("Hello, world!")
+    return {
+        "embeddings": "This is the embeddings endpoint",
+        "test": test
+        }
+
 @app.get("/")
 def greet_json():
     return {"Hello": "World! My key is: " + MY_KEY}

requirements.txt

@@ -1,3 +1,4 @@
 fastapi
 uvicorn[standard]
+langchain-huggingface