Update README.md

README.md

@@ -52,35 +52,22 @@ The reference provides the following functional *configurations*:
 
 ### Usage
 
-Prerequisites:
-- Install d-Matrix ML Tools: `pip install dmx-mltools`
-- clone this repo. `cd` to the cloned repo.
+Install d-Matrix ML Tools first: `pip install dmx-mltools`.
 
 ```python
->>> import os
->>> from mltools import dmx
->>> from transformers import pipeline
->>> import evaluate
->>> from datasets import load_dataset
+import os
+from mltools.dmx import pipeline
 
->>> pipe = pipeline(
->>>     "text-generation",
->>>     model="d-matrix/gpt2",
->>>     revision="gpt2-xl",
->>>     dmx_config="BASELINE",
->>>     use_auth_token=os.environ.get("HUGGING_FACE_HUB_TOKEN"),
->>>     trust_remote_code=True,
->>>     # device_map="auto",  # enabling model parallel on multi-GPU nodes
->>> )
->>> pipe.model = dmx.Model(
->>>     pipe.model, monkey_patched=False, hf=True, input_names=["input_ids", "labels"]
->>> )
+pipe = pipeline(
+    task="text-generation",
+    model="d-matrix/gpt2",
+    revision="temp-distilgpt2",
+    dmx_config="BASELINE",
+    trust_remote_code=True,
+    # device_map="auto",  # enabling model parallel on multi-GPU nodes
+)
 
->>> perplexity = evaluate.load("d-matrix/dmx_perplexity", module_type="metric")
->>> input_texts = load_dataset("ptb_text_only", "penn_treebank", split="test")["sentence"]
->>> results = perplexity.compute(model=pipe.model.body, references=input_texts)
->>> print(results)
-{'loss': 3.0451672077178955, 'perplexity': 21.0135440826416}
+results = pipe.evaluate(metric="d-matrix/perplexity", dataset="wikitext-2")
 ```
 
 ### Evaluation results