Update app.py

app.py

@@ -194,7 +194,15 @@ if (runModel=='1'):
     model.save_pretrained(model_path)
     tokenizer.save_pretrained(tokenizer_path)
 
+    # Check for specific files in the model directory
+    model_files = os.listdir(model_path)
+    model_files = [file for file in model_files if file.endswith(('pytorch_model.bin', 'config.json'))]
+    print("Specific files in model directory:", model_files)
     
+    # Check for specific files in the tokenizer directory
+    tokenizer_files = os.listdir(tokenizer_path)
+    tokenizer_files = [file for file in tokenizer_files if file.endswith(('tokenizer.json', 'vocab.txt'))]
+    print("Specific files in tokenizer directory:", tokenizer_files)
     
 else:
     print('Load Pre-trained')