radinplaid commited on
Commit
ba88b59
·
verified ·
1 Parent(s): b26fc1c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -7
README.md CHANGED
@@ -50,22 +50,25 @@ See the `eole` model configuration in this repository for further details and th
50
 
51
  You must install the Nvidia cuda toolkit first, if you want to do GPU inference.
52
 
53
- Next, install the `quickmt` python library and download the model:
54
 
55
  ```bash
56
  git clone https://github.com/quickmt/quickmt.git
57
  pip install ./quickmt/
58
-
59
- quickmt-model-download quickmt/quickmt-it-en ./quickmt-it-en
60
  ```
61
 
62
- Finally use the model in python:
63
 
64
  ```python
65
  from quickmt import Translator
66
-
67
- # Auto-detects GPU, set to "cpu" to force CPU inference
68
- t = Translator("./quickmt-it-en/", device="auto")
 
 
 
 
 
69
 
70
  # Translate - set beam size to 1 for faster speed (but lower quality)
71
  sample_text = "Lo studio è ancora in fase iniziale, come dichiarato cautelativamente dal dottor Ehud Ur, professore di medicina alla Dalhousie University di Halifax, Nuova Scozia, e direttore del dipartimento clinico e scientifico della Canadian Diabetes Association."
 
50
 
51
  You must install the Nvidia cuda toolkit first, if you want to do GPU inference.
52
 
53
+ Next, install the `quickmt` [python library](github.com/quickmt/quickmt).
54
 
55
  ```bash
56
  git clone https://github.com/quickmt/quickmt.git
57
  pip install ./quickmt/
 
 
58
  ```
59
 
60
+ Finally, use the model in python:
61
 
62
  ```python
63
  from quickmt import Translator
64
+ from huggingface_hub import snapshot_download
65
+
66
+ # Download Model (if not downloaded already) and return path to local model
67
+ # Device is either 'auto', 'cpu' or 'cuda'
68
+ t = Translator(
69
+ snapshot_download("quickmt/quickmt-it-en", ignore_patterns="eole-model/*"),
70
+ device="cpu"
71
+ )
72
 
73
  # Translate - set beam size to 1 for faster speed (but lower quality)
74
  sample_text = "Lo studio è ancora in fase iniziale, come dichiarato cautelativamente dal dottor Ehud Ur, professore di medicina alla Dalhousie University di Halifax, Nuova Scozia, e direttore del dipartimento clinico e scientifico della Canadian Diabetes Association."