Update InferenceConfig.py

InferenceConfig.py

@@ -57,9 +57,9 @@ class InferenceConfig:
 
     def find_model(self, model_list):
         print("weights", self.weights)
-        for model_name, model_path in enumerate(model_list):
-            print("Path", model_path, "->", model_name)
-            if model_path == self.weights:
+        for model_name in model_list:
+            print("Path", model_list[model_name], "->", model_name)
+            if model_list[model_name] == self.weights:
                 return model_name
         print("not found")
         return None