Update app.py

app.py

@@ -5,15 +5,18 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 model_repo_path = 'nxmwxm/correct_answer'
 
-# Load tokenizer and model
+# Load the tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_repo_path)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_repo_path)
 
-# Check and add special tokens if needed
+# Define special tokens
 special_tokens_dict = {'additional_special_tokens': ['<extra_id_99>']}
 num_added_toks = tokenizer.add_special_tokens(special_tokens_dict)
+
+# Load the model and resize token embeddings
+model = AutoModelForSeq2SeqLM.from_pretrained(model_repo_path)
 model.resize_token_embeddings(len(tokenizer))
 
+
 # Load the pipeline with your model and tokenizer
 qa_pipeline = pipeline(
     'question-answering',