Update app.py

app.py

@@ -40,6 +40,9 @@ def setup_training():
     # Print dataset structure
     logging.info(f"Dataset columns: {dataset['train'].column_names}")
 
+    # Temporarily return here to inspect the columns
+    return
+
     logging.info(f"Dataset loaded. Train size: {len(dataset['train'])}, Test size: {len(dataset['test'])}")
 
     # Load tokenizer and model