rudradcruze commited on
Commit
08c55d8
·
1 Parent(s): 1c25c67

change the total model required section

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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=4,
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),