Update app.py

app.py

@@ -6,7 +6,8 @@ from transformers import (
     AutoModelForCausalLM, 
     BitsAndBytesConfig, 
     TrainingArguments, 
-    Trainer
+    Trainer,
+    DataCollatorForLanguageModeling
 )
 from peft import LoraConfig, TaskType, get_peft_model, prepare_model_for_kbit_training
 from datasets import Dataset
@@ -306,7 +307,7 @@ def train_model(progress=gr.Progress()):
             model=model,
             args=training_args,
             train_dataset=train_dataset,
-            data_collator=transformers.DataCollatorForLanguageModeling(
+            data_collator=DataCollatorForLanguageModeling(
                 tokenizer=None, mlm=False
             )
         )