marcodambra commited on
Commit
3129cb0
1 Parent(s): 197f75b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -40,8 +40,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
40
 
41
  device = "cuda" # change to cpu if you have no gpu
42
 
43
- model = AutoModelForCausalLM.from_pretrained("MoxoffSpA/Pompei")
44
- tokenizer = AutoTokenizer.from_pretrained("MoxoffSpA/Pompei")
45
 
46
  messages = [
47
  {"role": "user", "content": "Qual è il tuo piatto preferito?"},
 
40
 
41
  device = "cuda" # change to cpu if you have no gpu
42
 
43
+ model = AutoModelForCausalLM.from_pretrained("MoxoffSpA/Azzurro")
44
+ tokenizer = AutoTokenizer.from_pretrained("MoxoffSpA/Azzurro")
45
 
46
  messages = [
47
  {"role": "user", "content": "Qual è il tuo piatto preferito?"},