liamcripwell commited on
Commit
43a2523
1 Parent(s): cae4db1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -17,9 +17,14 @@ It is recommended to use the [plan_simp](https://github.com/liamcripwell/plan_si
17
  Here is how to load this model in PyTorch:
18
 
19
  ```python
 
20
  from plan_simp.models.bart import load_simplifier
21
-
22
  simplifier, tokenizer, hparams = load_simplifier("liamcripwell/o-ledpara")
 
 
 
 
 
23
  ```
24
 
25
  Plan-guided generation and evaluation can be run from the terminal (see the repo for more details).
 
17
  Here is how to load this model in PyTorch:
18
 
19
  ```python
20
+ # loading
21
  from plan_simp.models.bart import load_simplifier
 
22
  simplifier, tokenizer, hparams = load_simplifier("liamcripwell/o-ledpara")
23
+
24
+ # generation
25
+ from plan_simp.scripts.generate import Launcher
26
+ launcher = Launcher()
27
+ launcher.dynamic(model_ckpt="liamcripwell/o-ledpara", clf_model_ckpt="liamcripwell/pgdyn-plan", **params)
28
  ```
29
 
30
  Plan-guided generation and evaluation can be run from the terminal (see the repo for more details).