Update app.py

app.py

@@ -6,14 +6,21 @@ genai.configure(api_key=token)
 
 # Chargez l'image
 model = genai.GenerativeModel(model_name="gemini-pro-vision")
+model_simple = genai.GenerativeModel(model_name="gemini-pro")
 
 e =""
 # Fonction pour générer le contenu
 def generate_content(pro,image):
     global e
-    response = model.generate_content([pro, image])
-    print(response.text)
-    e = response.text
+
+    if not image:
+        response = model.generate_content(pro)
+        e = response.text
+
+    else:
+        response = model.generate_content([pro, image])
+        print(response.text)
+        e = response.text
     return e