Update app.py

app.py

@@ -1,8 +1,18 @@
 import streamlit as st
 from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
 
-# Replace with your Hugging Face model repository path
-model_repo_path = 'nxmwxm/answer_generator'
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+model_repo_path = 'nxmwxm/correct_answer'
+
+# Load tokenizer and model
+tokenizer = AutoTokenizer.from_pretrained(model_repo_path)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_repo_path)
+
+# Check and add special tokens if needed
+special_tokens_dict = {'additional_special_tokens': ['<extra_id_99>']}
+num_added_toks = tokenizer.add_special_tokens(special_tokens_dict)
+model.resize_token_embeddings(len(tokenizer))
 
 # Load the pipeline with your model and tokenizer
 qa_pipeline = pipeline(