Fixed the message and status code;

toggle-corp · Oct 15, 2024 · 7e59990 · 7e59990
1 parent b7c5e23
commit 7e59990
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -2,7 +2,7 @@
 from typing import List, Optional, Union
 
 from dotenv import load_dotenv
-from fastapi import FastAPI
+from fastapi import FastAPI, Response, status
 from pydantic import BaseModel
 
 from embedding_models import (
@@ -37,7 +37,7 @@ class RequestSchemaForEmbeddings(BaseModel):
 
 @app.get("/")
 async def home():
-    return "Embedding handler using models for texts", 200
+    return Response(content="Embedding handler using models for texts", status_code=status.HTTP_200_OK)
 
 
 @app.post("/get_embeddings")