Woziii commited on
Commit
72dbc28
1 Parent(s): 00c29e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,7 +4,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
4
  import spaces
5
  import matplotlib.pyplot as plt
6
  import numpy as np
7
-
 
 
8
  # Liste des modèles
9
  models = [
10
  "meta-llama/Llama-2-13b", "meta-llama/Llama-2-7b", "meta-llama/Llama-2-70b",
 
4
  import spaces
5
  import matplotlib.pyplot as plt
6
  import numpy as np
7
+ from huggingface_hub import login
8
+ import os
9
+ login(token=os.environ["HF_TOKEN"])
10
  # Liste des modèles
11
  models = [
12
  "meta-llama/Llama-2-13b", "meta-llama/Llama-2-7b", "meta-llama/Llama-2-70b",