detect status

This commit is contained in:
josc146
2023-05-19 15:59:04 +08:00
parent db61d3f7f9
commit 7ba90ae7af
4 changed files with 28 additions and 4 deletions

View File

@@ -6,9 +6,9 @@ Model_Config = "model_config"
class ModelStatus(Enum):
Offline = auto()
Loading = auto()
Working = auto()
Offline = 0
Loading = 2
Working = 3
def init():

View File

@@ -59,3 +59,8 @@ def update_config(body: ModelConfigBody):
global_var.set(global_var.Model_Config, body)
return "success"
@router.get("/status")
def status():
return {"status": global_var.get(global_var.Model_Status)}