Create app.py

app.py

@@ -0,0 +1,23 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import List
+import torch
+from transformers import AutoTokenizer, AutoModel
+from src.model.encoder import ProdFeatureEncoder
+from src.config import ModelConfig
+
+app = FastAPI()
+
+class TextInput(BaseModel):
+    text: str
+
+class EmbeddingOutput(BaseModel):
+    embedding: List[float]
+
+@app.post("/encode_text", response_model=EmbeddingOutput)
+async def encode_text(text_input: TextInput):
+    config = ModelConfig()
+    model = ProdFeatureEncoder(config=config)
+    embedding = model(text_input.text)
+    return {"embedding": embedding.tolist()}
+