Update app.py

app.py

@@ -6,10 +6,8 @@ from transformers import (
     AutoTokenizer,
     DataCollatorWithPadding,
     HfArgumentParser,
-    PreTrainedModel,
-    PretrainedConfig,
     Trainer,
-    training_args,
+    TrainingArguments,
 )
 
 class MockOpenAI:
@@ -38,7 +36,7 @@ class MockOpenAI:
         self.data_collator = DataCollatorWithPadding(self.tokenizer)
         self.trainer = Trainer(
             model=self.model,
-            args=training_args(
+            args=TrainingArguments(
                 output_dir="./",
                 num_train_epochs=1,
                 learning_rate=1e-5,
@@ -49,11 +47,11 @@ class MockOpenAI:
         )
 
     class Chat:
-        def __init__(self, mock_openai: MockOpenAI):
+        def __init__(self, mock_openai: 'MockOpenAI'):
             self.mock_openai = mock_openai
 
         class Completions:
-            def __init__(self, mock_openai: MockOpenAI):
+            def __init__(self, mock_openai: 'MockOpenAI'):
                 self.mock_openai = mock_openai
 
             def create(