Spaces:
Running
on
Zero
Running
on
Zero
fix: weights
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def find_pth_files(directory):
|
|
| 53 |
return pth_files
|
| 54 |
|
| 55 |
def remove_weights_prefix(paths):
|
| 56 |
-
cleaned_paths = [path.replace('weights
|
| 57 |
return cleaned_paths
|
| 58 |
|
| 59 |
directory = "weights"
|
|
|
|
| 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"
|