Update app.py

app.py

@@ -3,10 +3,39 @@ import google.generativeai as genai
 
 import os 
 token=os.environ.get("TOKEN") 
-genai.configure(api_key=token)
-
-# Chargez l'image
-model = genai.GenerativeModel(model_name="gemini-1.5-pro-latest")
+e = ""
+genai.configure(
+    api_key=token
+)
+generation_config = {
+  "temperature": 1,
+  "top_p": 0.95,
+  "top_k": 64,
+  "max_output_tokens": 8192,
+}
+
+safety_settings = [
+  {
+    "category": "HARM_CATEGORY_HARASSMENT",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_HATE_SPEECH",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+    "threshold": "BLOCK_NONE"
+  },
+]
+
+model = genai.GenerativeModel(model_name="gemini-1.5-flash-latest",
+                              generation_config=generation_config,
+                              safety_settings=safety_settings)
 
 
 meto = """
@@ -77,19 +106,19 @@ mise = """
 
 
 
-e =""
+
 # Fonction pour générer le contenu
 def generate_content(image,choix):
     global e
 
 
     
-    if choix == "sans":
+    if choix == "sans mise en scène.":
         pro = f"traite moi ce devoir en respectant scrupuleusement la méthodologie suivante : {meto}"
         response = model.generate_content([pro, image])
         print(response.text)
         e = response.text
-    elif choix =="avec":
+    elif choix =="mise en scène uniquement":
         pre = f"traite moi ce devoir en respectant scrupuleusement la méthodologie suivante.je le veux  Sous forme de paragraphe. : {mise}"
         response = model.generate_content([pre, image])
         print(response.text)
@@ -102,7 +131,7 @@ def generate_content(image,choix):
 
 
 markdown = r"""
-e
+{e}
 """.format(e)
 # Interface Gradio
 iface = gr.Interface(fn=generate_content, inputs=[gr.Image(type='pil'),gr.Radio(["sans","avec"])] , outputs= gr.Markdown(markdown, latex_delimiters=[{ "left":"$$", "right":"$$", "display": True }]))