Kevin Fink commited on
Commit
e504e90
·
1 Parent(s): 77c210a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
40
 
41
  # Tokenize the dataset
42
  def tokenize_function(examples):
43
- max_length = 64
44
  # Assuming 'text' is the input and 'target' is the expected output
45
  model_inputs = tokenizer(
46
  examples['text'],
 
40
 
41
  # Tokenize the dataset
42
  def tokenize_function(examples):
43
+ max_length = 16
44
  # Assuming 'text' is the input and 'target' is the expected output
45
  model_inputs = tokenizer(
46
  examples['text'],