Update app.py
Browse files
app.py
CHANGED
|
@@ -40,9 +40,6 @@ def setup_training():
|
|
| 40 |
# Print dataset structure
|
| 41 |
logging.info(f"Dataset columns: {dataset['train'].column_names}")
|
| 42 |
|
| 43 |
-
# Temporarily return here to inspect the columns
|
| 44 |
-
return
|
| 45 |
-
|
| 46 |
logging.info(f"Dataset loaded. Train size: {len(dataset['train'])}, Test size: {len(dataset['test'])}")
|
| 47 |
|
| 48 |
# Load tokenizer and model
|
|
|
|
| 40 |
# Print dataset structure
|
| 41 |
logging.info(f"Dataset columns: {dataset['train'].column_names}")
|
| 42 |
|
|
|
|
|
|
|
|
|
|
| 43 |
logging.info(f"Dataset loaded. Train size: {len(dataset['train'])}, Test size: {len(dataset['test'])}")
|
| 44 |
|
| 45 |
# Load tokenizer and model
|