Update app.py

app.py

@@ -168,8 +168,12 @@ if (runModel=='1'):
     evaluate_and_report_errors(model,train_dataloader, tokenizer)
 
     # Save the model and tokenizer
-    model.save_pretrained('./'  + modelNameToUse + '_model')
-    tokenizer.save_pretrained('./' + modelNameToUse + '_tokenizer')
+    #model.save_pretrained('./'  + modelNameToUse + '_model')
+    #tokenizer.save_pretrained('./' + modelNameToUse + '_tokenizer')
+
+    model.save_pretrained("hf-data-timeframe/data-timeframe_model")
+    tokenizer.save_pretrained("hf-data-timeframe/data-timeframe_tokenizer")
+    
 else:
     print('Load Pre-trained')
     #model_save_path = "./" + modelNameToUse + "_model"