alex-abb commited on
Commit
67a27fd
·
verified ·
1 Parent(s): d393f74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import spaces
3
  import time
4
  import os
5
  key = (os.getenv('API_KEY'))
 
6
 
7
  # Utiliser le jeton d'accès API pour charger le modèle
8
  from transformers import pipeline
@@ -12,6 +13,7 @@ messages = [
12
  ]
13
  pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B-Instruct", token=key)
14
  pipe(messages)
 
15
 
16
 
17
 
 
3
  import time
4
  import os
5
  key = (os.getenv('API_KEY'))
6
+ from transformers import AutoTokenizer, AutoModelForCausalLM
7
 
8
  # Utiliser le jeton d'accès API pour charger le modèle
9
  from transformers import pipeline
 
13
  ]
14
  pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B-Instruct", token=key)
15
  pipe(messages)
16
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
17
 
18
 
19