pszemraj commited on
Commit
8b5e267
1 Parent(s): 99fc095

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -28,9 +28,14 @@ This model is a fine-tuned version of [sberbank-ai/mGPT](https://huggingface.co/
28
 
29
  Install the transformers library if you don't have it:
30
 
 
31
  pip install -U transformers
 
 
32
  load the model into a pipeline object:
33
 
 
 
34
  from transformers import pipeline
35
  import torch
36
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -38,6 +43,8 @@ my_chatbot = pipeline('text-generation',
38
  'pszemraj/mGPT-Peter-mwe',
39
  device=0 if device == 'cuda' else -1,
40
  )
 
 
41
 
42
 
43
 
 
28
 
29
  Install the transformers library if you don't have it:
30
 
31
+ ```
32
  pip install -U transformers
33
+ ```
34
+
35
  load the model into a pipeline object:
36
 
37
+
38
+ ```
39
  from transformers import pipeline
40
  import torch
41
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
43
  'pszemraj/mGPT-Peter-mwe',
44
  device=0 if device == 'cuda' else -1,
45
  )
46
+ ```
47
+
48
 
49
 
50