MisterAI commited on
Commit
0ff1405
·
verified ·
1 Parent(s): ab24fa5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -50,11 +50,15 @@ Analysez le texte suivant et créez une présentation claire et professionnelle
50
 
51
 
52
  # Téléchargement du modèle
 
 
 
 
53
  model_id = "mistralai/Mistral-Nemo-Instruct-2407"
54
 
55
  # Initialisation du modèle avec des paramètres de contexte plus grands
56
  tokenizer = AutoTokenizer.from_pretrained(model_id)
57
- model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
58
 
59
  def parse_presentation_content(content):
60
  """Parse le contenu généré en sections pour les diapositives"""
 
50
 
51
 
52
  # Téléchargement du modèle
53
+ # Authentification Necessaire
54
+ #https://huggingface.co/docs/hub/spaces-overview#managing-secrets
55
+ #https://huggingface.co/docs/hub/security-tokens
56
+ token = os.getenv('Authentification_HF')
57
  model_id = "mistralai/Mistral-Nemo-Instruct-2407"
58
 
59
  # Initialisation du modèle avec des paramètres de contexte plus grands
60
  tokenizer = AutoTokenizer.from_pretrained(model_id)
61
+ model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", token=token)
62
 
63
  def parse_presentation_content(content):
64
  """Parse le contenu généré en sections pour les diapositives"""