add: data generator
This commit is contained in:
6
app.py
6
app.py
@@ -56,10 +56,13 @@ def train(request: TrainRequest, background_tasks: BackgroundTasks):
|
||||
def try_load(request: TrainRequest, background_tasks: BackgroundTasks):
|
||||
global solver
|
||||
global status
|
||||
if status != "initialized":
|
||||
return {"Status": "model not initialized"}
|
||||
if solver.load_model():
|
||||
status = "ready"
|
||||
return {"Status": "ok"}
|
||||
background_tasks.add_task(lambda : task(request))
|
||||
else:
|
||||
background_tasks.add_task(lambda : task(request))
|
||||
|
||||
|
||||
class BoardRequest(BaseModel):
|
||||
@@ -115,7 +118,6 @@ def load():
|
||||
return {"Status": "model loading failed"}
|
||||
|
||||
|
||||
|
||||
@app.get("/status")
|
||||
def get_status():
|
||||
global status
|
||||
|
||||
Reference in New Issue
Block a user