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 |
|