Update app.py

app.py

@@ -47,16 +47,18 @@ class MockOpenAI:
         self.model = AutoModelForSequenceClassification.from_pretrained(self.model_name, config=self.config)
         self.data_collator = DataCollatorWithPadding(self.tokenizer)
         self.trainer = Trainer(
-            model=self.model,
-            args=TrainingArguments(
-                output_dir="./",
-                num_train_epochs=1,
-                learning_rate=1e-5,
-                per_device_train_batch_size=16,
-                per_device_eval_batch_size=16,
-                evaluation_strategy="epoch",
-            ),
-        )
+        model=self.model,
+        args=TrainingArguments(
+           output_dir="./",
+           num_train_epochs=1,
+           learning_rate=1e-5,
+           per_device_train_batch_size=16,
+           per_device_eval_batch_size=16,
+           evaluation_strategy="epoch",
+           # Remove this line:
+           # use_seedable_sampler=True, 
+       ),
+   )
 
     class Chat:
         def __init__(self, mock_openai: 'MockOpenAI'):