update

main.py

@@ -1,9 +1,3 @@
-from pydantic import BaseModel
-
-class Request(BaseModel):
-    body: str
-
-
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
@@ -13,7 +7,7 @@ from transformers import pipeline
 app = FastAPI()
 
 @app.post("/query_gorilla")
-def query_gorilla(req: Request):
+def query_gorilla(req: str = Body(..., embed=True)):
     return str(req)
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")

requirements.txt

@@ -6,4 +6,3 @@ transformers==4.*
 uvicorn[standard]==0.17.*
 protobuf==4.*
 accelerate==0.27.*
-pydantic