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