CosmoAI commited on
Commit
4ac96da
1 Parent(s): a7ac55a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -165,7 +165,7 @@ def prepare_training_data(qa_pairs, tokenizer):
165
  return encoded_data
166
 
167
  # Prepare training data
168
- train_data = prepare_training_data(qa_pairs)
169
 
170
  # Train the model
171
  learning_rate = 2e-5
 
165
  return encoded_data
166
 
167
  # Prepare training data
168
+ train_data = prepare_training_data(qa_pairs, tokenizer)
169
 
170
  # Train the model
171
  learning_rate = 2e-5