Docfile commited on
Commit
e44c6ac
·
verified ·
1 Parent(s): 22f8616

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -4,9 +4,15 @@ import os
4
 
5
  # Configurez l'API Gemini
6
  genai.configure(api_key=os.environ["GEMINI_API_KEY"]) # Remplacez "VOTRE_CLÉ_API" par votre clé API Gemini
 
 
 
 
 
 
7
 
8
  # Définissez le modèle
9
- model = genai.GenerativeModel("gemini-2.0-flash-exp")
10
 
11
  # Chemin vers le fichier PDF (assurez-vous qu'il est correct)
12
  PDF_FILE_PATH = "okoumba-nkoghe-elo_compress.pdf"
 
4
 
5
  # Configurez l'API Gemini
6
  genai.configure(api_key=os.environ["GEMINI_API_KEY"]) # Remplacez "VOTRE_CLÉ_API" par votre clé API Gemini
7
+ safety_settings = [
8
+ {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
9
+ {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
10
+ {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
11
+ {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
12
+ ]
13
 
14
  # Définissez le modèle
15
+ model = genai.GenerativeModel("gemini-2.0-flash-exp",safety_settings=safety_settings)
16
 
17
  # Chemin vers le fichier PDF (assurez-vous qu'il est correct)
18
  PDF_FILE_PATH = "okoumba-nkoghe-elo_compress.pdf"