yifan0sun commited on
Commit
5e5b29d
·
1 Parent(s): c0adfb1

dockerfile

Browse files
Files changed (1) hide show
  1. server.py +5 -1
server.py CHANGED
@@ -20,7 +20,7 @@ app = FastAPI()
20
 
21
  app.add_middleware(
22
  CORSMiddleware,
23
- allow_origins=["*"], # or restrict to ["http://localhost:3000"]
24
  allow_credentials=True,
25
  allow_methods=["*"],
26
  allow_headers=["*"],
@@ -52,6 +52,10 @@ class PredModelRequest(BaseModel):
52
  hypothesis:str
53
  maskID: int | None = None
54
 
 
 
 
 
55
 
56
  @app.post("/load_model")
57
  def load_model(req: LoadModelRequest):
 
20
 
21
  app.add_middleware(
22
  CORSMiddleware,
23
+ allow_origins=["*"],
24
  allow_credentials=True,
25
  allow_methods=["*"],
26
  allow_headers=["*"],
 
52
  hypothesis:str
53
  maskID: int | None = None
54
 
55
+ @app.get("/ping")
56
+ def ping():
57
+ return {"message": "pong"}
58
+
59
 
60
  @app.post("/load_model")
61
  def load_model(req: LoadModelRequest):