Spaces:

shorecode
/

gradio-3

Sleeping

Kevin Fink commited on Dec 8, 2024

Commit

18b8b87

1 Parent(s): 6e42911

dev

Files changed (1) hide show

app.py CHANGED Viewed

@@ -209,7 +209,7 @@ 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['text'])
         print('DONE')

         dataset['validation'] = dataset['validation'].select(range(100))
         train_set = tokenize_function(dataset['train'])
         valid_set = tokenize_function(dataset['validation'])
+        print(valid_set)
         print('DONE')