Spaces:
Sleeping
Sleeping
from fastapi import FastAPI | |
from os import getenv | |
from langchain_huggingface import HuggingFaceEmbeddings | |
from fastapi.responses import HTMLResponse | |
from fastapi.staticfiles import StaticFiles | |
app = FastAPI() | |
MY_KEY = getenv("MY_KEY") | |
embeddings = HuggingFaceEmbeddings(model_name="jinaai/jina-embeddings-v2-small-en") | |
app.mount("/static", StaticFiles(directory="static"), name="static") | |
def get_embeddings(input: str): | |
result = embeddings.embed_query(input) | |
return { | |
"embeddings": result, | |
"test": "testtext" | |
} | |
def get_index(): | |
with open("static/index.html") as f: | |
return f.read() | |