Reyad-Ahmmed commited on
Commit
ee7b10d
·
verified ·
1 Parent(s): 08d4674

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -312,8 +312,8 @@ if (should_train_model=='1'): #train model
312
  model.config.label_mapping = label_mapping
313
 
314
  # Save the model and tokenizer
315
- model.save_pretrained(f"./{model_save_path}")
316
- tokenizer.save_pretrained('./saved_fleet_tokenizer')
317
 
318
  #for push repository
319
  repo_name = "Reyad-Ahmmed/hf-data-timeframe"
 
312
  model.config.label_mapping = label_mapping
313
 
314
  # Save the model and tokenizer
315
+ model.save_pretrained(f"./{model_save_path}_model")
316
+ tokenizer.save_pretrained(f"./{model_save_path}_tokenizer")
317
 
318
  #for push repository
319
  repo_name = "Reyad-Ahmmed/hf-data-timeframe"