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