Update app.py

app.py

@@ -1,12 +1,17 @@
+import os
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-# Charger le modèle
+# Récupérer le token depuis les secrets du Space
+hf_token = os.getenv("HF_TOKEN")
+
+# Charger le modèle avec le token
 model_name = "mistralai/Mistral-7B-Instruct-v0.2"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
+    use_auth_token=hf_token,
     torch_dtype=torch.float16,
     device_map="auto"
 )
@@ -30,4 +35,4 @@ with gr.Blocks() as demo:
     msg = gr.Textbox(placeholder="Écris un message...")
     msg.submit(chat, [msg, chatbot], [chatbot, chatbot])
 
-demo.launch()
+demo.launch()