tokenizer

src/streamlit_app.py

@@ -71,7 +71,7 @@ def preprocess_function(examples, tokenizer):
     return {key: tensor.to(device) for key, tensor in inputs.items()}
 
 def generate_q(context, answer):
-    tokenizer = AutoTokenizer.from_pretrained('VietAI/vit5-base')
+    tokenizer = AutoTokenizer.from_pretrained('shnl/inst-qg-vinewsqa-vit5')
     model = AutoModelForSeq2SeqLM.from_pretrained('shnl/inst-qg-vinewsqa-vit5')
     instruction_qg = 'Hãy tạo ra câu hỏi từ nội dung của ngữ cảnh và câu trả lời dưới đây.',
     examples = {"input_seq": str("### Instruction: \n"