dev

app.py

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