hgrrg commited on
Commit
ee03616
·
verified ·
1 Parent(s): 6bb2686

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -7,8 +7,7 @@ tags:
7
 
8
  Wanna some meos ? Run this
9
 
10
- ```
11
- python
12
  from transformers import AutoTokenizer, AutoModelForCausalLM
13
  import torch
14
 
@@ -19,7 +18,7 @@ tokenizer = AutoTokenizer.from_pretrained("./meomeo", local_files_only=True)
19
  model = AutoModelForCausalLM.from_pretrained("./meomeo", local_files_only=True)
20
 
21
  # Simple inference
22
- text = "Once upon a time,"
23
  inputs = tokenizer(text, return_tensors="pt")
24
  outputs = model.generate(**inputs, max_length=50)
25
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 
7
 
8
  Wanna some meos ? Run this
9
 
10
+ ```python
 
11
  from transformers import AutoTokenizer, AutoModelForCausalLM
12
  import torch
13
 
 
18
  model = AutoModelForCausalLM.from_pretrained("./meomeo", local_files_only=True)
19
 
20
  # Simple inference
21
+ text = "Mèo Méo Meo Mèo Meo"
22
  inputs = tokenizer(text, return_tensors="pt")
23
  outputs = model.generate(**inputs, max_length=50)
24
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))