commit

model_handler.py

@@ -25,7 +25,7 @@ class ModelHandler:
             json.dump(self.model_infos, f, indent=4)
 
     def get_arm_bench_data(self):
-        models = self.api.list_models(filter="Arm-LLM-Benchmark")
+        models = self.api.list_models(filter="ArmBench-LLM")
         model_names = {model["model_name"] for model in self.model_infos}
         repositories = [model.modelId for model in models]