Spaces:
Running
on
Zero
Running
on
Zero
fix: weights
Browse files
app.py
CHANGED
@@ -53,10 +53,10 @@ def find_pth_files(directory):
|
|
53 |
return pth_files
|
54 |
|
55 |
def remove_weights_prefix(paths):
|
56 |
-
cleaned_paths = [path.replace('
|
57 |
return cleaned_paths
|
58 |
|
59 |
-
directory = "
|
60 |
pth_files = find_pth_files(directory)
|
61 |
pth_files2 = remove_weights_prefix(pth_files)
|
62 |
|
|
|
53 |
return pth_files
|
54 |
|
55 |
def remove_weights_prefix(paths):
|
56 |
+
cleaned_paths = [path.replace('weights\\', '') for path in paths]
|
57 |
return cleaned_paths
|
58 |
|
59 |
+
directory = "weights"
|
60 |
pth_files = find_pth_files(directory)
|
61 |
pth_files2 = remove_weights_prefix(pth_files)
|
62 |
|