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

fix: weights

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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('.\\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
 
 
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