Reyad-Ahmmed commited on
Commit
04356f9
·
verified ·
1 Parent(s): cfa2743

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -196,12 +196,12 @@ if (runModel=='1'):
196
 
197
  # Check for specific files in the model directory
198
  model_files = os.listdir(model_path)
199
- model_files = [file for file in model_files if file.endswith(('pytorch_model.bin', 'config.json'))]
200
  print("Specific files in model directory:", model_files)
201
 
202
  # Check for specific files in the tokenizer directory
203
  tokenizer_files = os.listdir(tokenizer_path)
204
- tokenizer_files = [file for file in tokenizer_files if file.endswith(('tokenizer.json', 'vocab.txt'))]
205
  print("Specific files in tokenizer directory:", tokenizer_files)
206
 
207
  else:
 
196
 
197
  # Check for specific files in the model directory
198
  model_files = os.listdir(model_path)
199
+ model_files = [file for file in model_files]
200
  print("Specific files in model directory:", model_files)
201
 
202
  # Check for specific files in the tokenizer directory
203
  tokenizer_files = os.listdir(tokenizer_path)
204
+ tokenizer_files = [file for file in tokenizer_files]
205
  print("Specific files in tokenizer directory:", tokenizer_files)
206
 
207
  else: