Update app.py

app.py

@@ -1,11 +1,11 @@
-from transformers import Trainer, TrainingArguments, DeepSeekForQuestionAnswering, DeepSeekTokenizer
+from transformers import Trainer, TrainingArguments, AutoModelForQuestionAnswering, AutoTokenizer
 from datasets import load_dataset
 
-# Load the DeepSeek model and tokenizer
-model = DeepSeekForQuestionAnswering.from_pretrained("DeepSeek/DeepSeek-v3")
-tokenizer = DeepSeekTokenizer.from_pretrained("DeepSeek/DeepSeek-v3")
+# Load the model and tokenizer from the Hub
+model = AutoModelForQuestionAnswering.from_pretrained("DeepSeek/DeepSeek-v3")
+tokenizer = AutoTokenizer.from_pretrained("DeepSeek/DeepSeek-v3")
 
-# Load dataset
+# Load your dataset
 dataset = load_dataset("json", data_files={"train": "your_dataset_train.json", "test": "your_dataset_test.json"})
 
 # Tokenize the dataset
@@ -25,7 +25,7 @@ training_args = TrainingArguments(
     weight_decay=0.01
 )
 
-# Initialize Trainer
+# Initialize the Trainer
 trainer = Trainer(
     model=model,
     args=training_args,
@@ -33,7 +33,7 @@ trainer = Trainer(
     eval_dataset=tokenized_datasets['test']
 )
 
-# Start the fine-tuning
+# Start fine-tuning
 trainer.train()
 
 # Save the model after fine-tuning