Update app.py

app.py

@@ -321,7 +321,8 @@ dataset.set_format(type="torch", columns=["input_ids", "labels"])
 tf_dataset = model.prepare_tf_dataset(
     dataset,
     batch_size=4,
-    shuffle=True,)
+    shuffle=True,
+    trust_remote_code=True)
 
 # %% [markdown]
 # **6**. Start training with your machine learning framework! Check out the 🤗 Transformers [text classification guide](https://huggingface.co/docs/transformers/tasks/sequence_classification) for an end-to-end example of how to train a model on a text dataset.