Spaces:
Runtime error
Runtime error
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"
|