Spaces:
Sleeping
Sleeping
dockerfile
Browse files
server.py
CHANGED
@@ -20,7 +20,7 @@ app = FastAPI()
|
|
20 |
|
21 |
app.add_middleware(
|
22 |
CORSMiddleware,
|
23 |
-
allow_origins=["*"],
|
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):
|