Update app.py

app.py

@@ -32,6 +32,7 @@ data_collator = DataCollatorForLanguageModeling(
     mlm=False
 )
 
+# Fine-tune the model
 # Fine-tune the model
 training_args = TrainingArguments(
     output_dir=output_dir,
@@ -40,10 +41,12 @@ training_args = TrainingArguments(
     per_device_train_batch_size=2,
     save_steps=10_000,
     save_total_limit=2,
-    logging_dir=output_dir,  # Add this line for logging
-    logging_steps=100,  # Adjust this value based on your requirements
+    logging_dir=output_dir,
+    logging_steps=100,
+    report_to=[],
 )
 
+
 trainer = Trainer(
     model=model,
     args=training_args,