luulinh90s commited on
Commit
806aaf0
·
1 Parent(s): 46c86e4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -155,7 +155,7 @@ def experiment(username, sample_index, seed, filename):
155
  absolute_dir_path = os.path.abspath(dir_path)
156
  logger.info(f"Checking directory: {absolute_dir_path} for file: {visualization_file}")
157
  if visualization_file.strip() in os.listdir(absolute_dir_path):
158
- visualization_path = os.path.join(vis_dir, category, visualization_file)
159
  break
160
 
161
  if not visualization_path:
 
155
  absolute_dir_path = os.path.abspath(dir_path)
156
  logger.info(f"Checking directory: {absolute_dir_path} for file: {visualization_file}")
157
  if visualization_file.strip() in os.listdir(absolute_dir_path):
158
+ visualization_path = os.path.join(dir_path, visualization_file)
159
  break
160
 
161
  if not visualization_path: