Update app.py

app.py

@@ -5,24 +5,15 @@ from Similarity import Similarity
 app = FastAPI()
 similarity_model = Similarity()
 
-# Define the Pydantic model BEFORE using it
-class TextPairRequest(BaseModel):
+class TextRequest(BaseModel):
     text: str
-    claim: str
 
-@app.post("/get_sim_text")
-def get_sim_text_endpoint(request: TextPairRequest):
+@app.post("/embed")
+def embed_text_endpoint(request: TextRequest):
     try:
-        similarity_model.load_model()  # ensure model is loaded
-        claim_embedding = similarity_model.model.encode(
-            request.claim,
-            convert_to_tensor=True,
-            show_progress_bar=False
-        )
-        results = similarity_model.get_sim_text(
-            request.text,
-            claim_embedding
-        )
-        return {"result": results}
+        embedding = similarity_model.embed_text(request.text)
+        # Convert tensor to list so it can be JSON serialized
+        embedding_list = embedding.cpu().tolist() if hasattr(embedding, 'cpu') else embedding.tolist()
+        return {"embedding": embedding_list}
     except Exception as e:
         return {"error": str(e)}