Update app.py

app.py

@@ -110,7 +110,7 @@ if (runModel=='1'):
     api = HfApi()
     create_repo(repo_id=repo_name, token=api_token, exist_ok=True)
     
-           
+    
     # Create an instance of the custom loss function
     training_args = TrainingArguments(
         output_dir=f'{repo_name}/results_' + modelNameToUse,
@@ -198,7 +198,11 @@ if (runModel=='1'):
     # Save the model and tokenizer locally
     local_model_path = "./data-timeframe_model"
     local_tokenizer_path = "./data-timeframe_tokenizer"
-
+    
+    # Ensure the directories exist 
+    os.makedirs(local_model_path, exist_ok=True) 
+    os.makedirs(local_tokenizer_path, exist_ok=True)
+    
     model.save_pretrained(local_model_path)
     tokenizer.save_pretrained(local_tokenizer_path)