Spaces:

shorecode
/

gradio-3

Sleeping

Kevin Fink commited on Dec 9, 2024

Commit

ed395a5

1 Parent(s): efbd624

deve

Files changed (1) hide show

app.py CHANGED Viewed

@@ -210,7 +210,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
         train_set = dataset.map(tokenize_function, batched=True)
         #valid_set = dataset['validation'].map(tokenize_function, batched=True)
-        print(train_set['labels'])
         #print(train_set.keys())
         print('DONE')

         train_set = dataset.map(tokenize_function, batched=True)
         #valid_set = dataset['validation'].map(tokenize_function, batched=True)
         #print(train_set.keys())
         print('DONE')