cors middleware

main.py

@@ -5,6 +5,7 @@ from pydantic.generics import GenericModel
 from sentence_transformers import SentenceTransformer
 from fastapi import FastAPI
 import os, asyncio, numpy, ujson
+from fastapi.middleware.cors import CORSMiddleware
 
 MODEL = SentenceTransformer("all-mpnet-base-v2")
 
@@ -31,10 +32,6 @@ async def encode(sentences: List[str]) -> List[numpy.ndarray]:
     result = await loop.run_in_executor(None, _encode, sentences)
     return result
 
-class SemanticSearchReq(BaseModel):
-    query: str
-    candidates: List[str]
-
 class EmbedReq(BaseModel):
     sentences: List[str]
 
@@ -44,4 +41,11 @@ app = FastAPI()
 async def embed(embed: EmbedReq):
     result = await encode(embed.sentences)
     # Convert it to an ordinary list of floats
-    return ujson.dumps([r.tolist() for r in result])
+    return ujson.dumps([r.tolist() for r in result])
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)