add: data generator

This commit is contained in:
h z
2025-01-29 21:24:10 +00:00
parent 34e9b95145
commit 05d375d521
5 changed files with 101 additions and 7 deletions

6
app.py
View File

@@ -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