Spaces:

Reyad-Ahmmed
/

HF_Python

Sleeping

Reyad-Ahmmed commited on Jan 2

Commit

a57bfeb

verified ·

1 Parent(s): a2c2250

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ if (runModel=='1'):
     # Create an instance of the custom loss function
     training_args = TrainingArguments(
-        output_dir='./results_' + modelNameToUse,
         num_train_epochs=2,
         per_device_train_batch_size=2,
         per_device_eval_batch_size=2,

     # Create an instance of the custom loss function
     training_args = TrainingArguments(
+        output_dir=f'{repo_name}/results_' + modelNameToUse,
         num_train_epochs=2,
         per_device_train_batch_size=2,
         per_device_eval_batch_size=2,