Spaces:
Runtime error
Runtime error
Commit
·
cf87b0c
1
Parent(s):
c376f3c
Update InferenceConfig.py
Browse files- InferenceConfig.py +3 -0
InferenceConfig.py
CHANGED
@@ -56,9 +56,12 @@ class InferenceConfig:
|
|
56 |
self.byte_high_conf = high
|
57 |
|
58 |
def find_model(self, model_list):
|
|
|
59 |
for model_name, model_path in enumerate(model_list):
|
|
|
60 |
if model_path == self.weights:
|
61 |
return model_name
|
|
|
62 |
return None
|
63 |
|
64 |
|
|
|
56 |
self.byte_high_conf = high
|
57 |
|
58 |
def find_model(self, model_list):
|
59 |
+
print("weights", self.weights)
|
60 |
for model_name, model_path in enumerate(model_list):
|
61 |
+
print("Path", model_path, "->", model_name)
|
62 |
if model_path == self.weights:
|
63 |
return model_name
|
64 |
+
print("not found")
|
65 |
return None
|
66 |
|
67 |
|