Initial commit with FastAPI backend

README.md

@@ -3,7 +3,7 @@ title: MentalbertEmbedder
 emoji: 🐠
 colorFrom: pink
 colorTo: green
-sdk: gradio
+sdk: fastapi
 sdk_version: 5.24.0
 app_file: app.py
 pinned: false

app.py

@@ -0,0 +1,16 @@
+from fastapi import FastAPI, Request
+from pydantic import BaseModel
+from sentence_transformers import SentenceTransformer
+import torch
+
+app = FastAPI()
+
+model = SentenceTransformer("mental/mental-bert-base-uncased")  
+
+class InputText(BaseModel):
+    text: str
+
+@app.post("/embed")
+async def embed_text(input: InputText):
+    embedding = model.encode(input.text).tolist()
+    return {"embedding": embedding}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+sentence-transformers
+torch