Shobhank-iiitdwd commited on
Commit
2a684f1
1 Parent(s): 54a7a7d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -65,7 +65,7 @@ Secondly, we have performed task-specific distillation with [roberta-base-squad2
65
  ```python
66
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
67
 
68
- model_name = "deepset/tinyroberta-squad2"
69
 
70
  # a) Get predictions
71
  nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
 
65
  ```python
66
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
67
 
68
+ model_name = "Shobhank-iiitdwd/Distiled-roberta-squad2-QA"
69
 
70
  # a) Get predictions
71
  nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)