bambadij commited on
Commit
537dcf5
·
1 Parent(s): 672dbed
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import os
6
  # Accéder à une variable d'environnement
7
  access_token = os.getenv('MODEL_Token')
8
  # Load model directly
9
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-Nemo-Instruct-2407",token=access_token)
10
- model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-Nemo-Instruct-2407",token=access_token)
11
 
12
 
13
  default_prompt = """Bonjour,
 
6
  # Accéder à une variable d'environnement
7
  access_token = os.getenv('MODEL_Token')
8
  # Load model directly
9
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1",token=access_token)
10
+ model = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1",token=access_token)
11
 
12
 
13
  default_prompt = """Bonjour,