Update README.md

README.md

@@ -61,8 +61,8 @@ This model is designed for extractive question answering tasks where:
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 
 # Load model and tokenizer
-model = AutoModelForQuestionAnswering.from_pretrained("your-username/bert-squad-baseline")
-tokenizer = AutoTokenizer.from_pretrained("your-username/bert-squad-baseline")
+model = AutoModelForQuestionAnswering.from_pretrained("G20-CS4248/bert-baseline-qa")
+tokenizer = AutoTokenizer.from_pretrained("G20-CS4248/bert-baseline-qa")
 
 # Create QA pipeline
 qa_pipeline = pipeline(
@@ -99,8 +99,8 @@ Confidence: 0.9234
 import torch
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer
 
-model = AutoModelForQuestionAnswering.from_pretrained("your-username/bert-squad-baseline")
-tokenizer = AutoTokenizer.from_pretrained("your-username/bert-squad-baseline")
+model = AutoModelForQuestionAnswering.from_pretrained("G20-CS4248/bert-baseline-qa")
+tokenizer = AutoTokenizer.from_pretrained("G20-CS4248/bert-baseline-qa")
 
 question = "What is the capital of France?"
 context = "Paris is the capital and largest city of France."