Fediory commited on
Commit
d0fe588
·
1 Parent(s): be1e53a

fix: weights

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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\\', '') for path in paths]
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"