deve

app.py

@@ -226,8 +226,8 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
         trainer = Trainer(
             model=model,
             args=training_args,
-            train_dataset=train_set['train'], 
-            eval_dataset=train_set['validation'], 
+            train_dataset=dataset['train'], 
+            eval_dataset=dataset['validation'], 
             compute_metrics=compute_metrics,
             #data_collator=data_collator,
             processing_class=tokenizer,