dev

app.py

@@ -209,7 +209,8 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
         dataset['validation'] = dataset['validation'].select(range(100))
         train_set = tokenize_function(dataset['train'])
         valid_set = tokenize_function(dataset['validation'])
-        print(valid_set)
+        
+        print(train_set)
 
         print('DONE')
 
@@ -219,7 +220,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
             model=model,
             args=training_args,
             train_dataset=train_set, 
-            eval_dataset=train_set,
+            eval_dataset=valid_set,
             compute_metrics=compute_metrics,
             data_collator=data_collator,
             #processing_class=tokenizer,