Change usage section

README.md

@@ -23,25 +23,25 @@ This enables to specify:
 ## Usage
 
 The model is instantiated the same way as in the Transformers library.
-The only difference is that there are a few new training arguments specific to HPUs:
-
-```
-from optimum.habana import GaudiTrainer, GaudiTrainingArguments
-from transformers import RobertaModel, RobertaTokenizer
-
-tokenizer = RobertaTokenizer.from_pretrained("roberta-base")
-model = RobertaModel.from_pretrained("roberta-base")
-args = GaudiTrainingArguments(
-    output_dir="/tmp/output_dir",
-    use_habana=True,
-    use_lazy_mode=True,
-    gaudi_config_name="Habana/roberta-base",
-)
-
-trainer = GaudiTrainer(
-    model=model,
-    args=args,
-    tokenizer=tokenizer,
-)
-trainer.train()
+The only difference is that there are a few new training arguments specific to HPUs.
+
+[Here](https://github.com/huggingface/optimum-habana/blob/main/examples/question-answering/run_qa.py) is a question-answering example script to fine-tune a model on SQuAD. You can run it with RoBERTa with the following command:
+```bash
+python run_qa.py \
+  --model_name_or_path roberta-base \
+  --gaudi_config_name Habana/roberta-base \
+  --dataset_name squad \
+  --do_train \
+  --do_eval \
+  --per_device_train_batch_size 12 \
+  --per_device_eval_batch_size 8 \
+  --learning_rate 3e-5 \
+  --num_train_epochs 2 \
+  --max_seq_length 384 \
+  --output_dir /tmp/squad/ \
+  --use_habana \
+  --use_lazy_mode \
+  --throughput_warmup_steps 2
 ```
+
+Check the [documentation](https://huggingface.co/docs/optimum/habana/index) out for more advanced usage and examples.