Update app.py

app.py

@@ -3,14 +3,17 @@ import spaces
 ## Load model directly
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import HfApi
 
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
+# Récupérer le jeton d'accès API depuis Hugging Face Secrets
+api = HfApi()
+secret = api.secrets.get("alex-abb/8B-on-GPU-ZERO/API_KEY")
 
+# Utiliser le jeton d'accès API pour charger le modèle
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", token=secret["value"])
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", token=secret["value"])
 
 
-print(API_KEY)
-