Commit
·
08c55d8
1
Parent(s):
1c25c67
change the total model required section
Browse files
app.py
CHANGED
|
@@ -652,6 +652,7 @@ async def predict_batch(request: BatchSequenceRequest):
|
|
| 652 |
}
|
| 653 |
)
|
| 654 |
|
|
|
|
| 655 |
@app.get("/example", response_model=PredictionResponse)
|
| 656 |
async def predict_example():
|
| 657 |
"""
|
|
@@ -758,6 +759,7 @@ async def predict_example():
|
|
| 758 |
}
|
| 759 |
)
|
| 760 |
|
|
|
|
| 761 |
@app.get("/health", response_model=HealthResponse)
|
| 762 |
async def health_check():
|
| 763 |
models_loaded = all([
|
|
@@ -801,7 +803,7 @@ async def health_check():
|
|
| 801 |
model_version=MODEL_VERSION,
|
| 802 |
models_loaded=models_loaded,
|
| 803 |
models_loaded_count=sum(1 for source in model_sources.values() if source["loaded"]),
|
| 804 |
-
total_models_required=
|
| 805 |
model_sources=model_sources,
|
| 806 |
repository_info=repository_info,
|
| 807 |
device=str(device),
|
|
|
|
| 652 |
}
|
| 653 |
)
|
| 654 |
|
| 655 |
+
|
| 656 |
@app.get("/example", response_model=PredictionResponse)
|
| 657 |
async def predict_example():
|
| 658 |
"""
|
|
|
|
| 759 |
}
|
| 760 |
)
|
| 761 |
|
| 762 |
+
|
| 763 |
@app.get("/health", response_model=HealthResponse)
|
| 764 |
async def health_check():
|
| 765 |
models_loaded = all([
|
|
|
|
| 803 |
model_version=MODEL_VERSION,
|
| 804 |
models_loaded=models_loaded,
|
| 805 |
models_loaded_count=sum(1 for source in model_sources.values() if source["loaded"]),
|
| 806 |
+
total_models_required=3,
|
| 807 |
model_sources=model_sources,
|
| 808 |
repository_info=repository_info,
|
| 809 |
device=str(device),
|