Update app.py

app.py

@@ -173,9 +173,24 @@ if (runModel=='1'):
     train_dataloader = DataLoader(train_dataset, batch_size=10, shuffle=True)
     evaluate_and_report_errors(model,train_dataloader, tokenizer)
 
+    model_path = './'  + modelNameToUse + '_model'
+    tokenizer_path = './' + modelNameToUse + '_tokenizer'
+
+    if os.path.isdir(model_path) && os.path.isdir(tokenizer_path):
+        print(f"The directory of model {model_path} exists.")
+        print("Directory contents:", os.listdir(model_path))
+
+        print(f"The directory of tokenizer{tokenizer_path} exists.")
+        print("Directory contents:", os.listdir(tokenizer_path))
+    else:
+        print(f"The directory {path} does not exist. Creating it now...")
+        os.makedirs(model_path, exist_ok=True)  # Create the directory
+        os.makedirs(tokenizer_path, exist_ok=True)  # Create the directory
+        print(f"Directory {path} created successfully.")
+        
     # Save the model and tokenizer
-    model.save_pretrained('./'  + modelNameToUse + '_model')
-    tokenizer.save_pretrained('./' + modelNameToUse + '_tokenizer')
+    model.save_pretrained(model_path)
+    tokenizer.save_pretrained(tokenizer_path)