Spaces:
Paused
Paused
Update app.py
Browse files
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(
|
| 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"
|