Update app.py

app.py

@@ -38,7 +38,7 @@ def process(model_id, dataset):
 
     # Step 4: Apply tokenization to dataset
 
-    tokenized_imdb = dataset_imdb.map(tokenizer(examples["text"], padding="max_length", truncation=True), batched=True)
+    tokenized_imdb = dataset_imdb.map(tokenizer(dataset_imdb["text"], padding="max_length", truncation=True), batched=True)
 
     #Step 5: Fine-tune the model