init

app.py

@@ -39,7 +39,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
     
         # Tokenize the dataset
         def tokenize_function(examples):
-            max_length = 64
+            max_length = 32
             return tokenizer(examples['text'], max_length=max_length, truncation=True)
     
         tokenized_datasets = dataset.map(tokenize_function, batched=True)
@@ -72,7 +72,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
             args=training_args,
             train_dataset=tokenized_datasets['train'],
             eval_dataset=tokenized_datasets['test'],
-            callbacks=[LoggingCallback()], 
+            #callbacks=[LoggingCallback()], 
         )
     
         # Fine-tune the model