Update app.py

app.py

@@ -113,15 +113,13 @@ aiplatform.init(project=project_id, location=location, credentials=credentials)
 
 # Función para generar imágenes con Vertex AI
 def generar_imagen_vertex(prompt):
-    model = aiplatform.TextToImage(model="projects/cloud-ml/models/gemini-1.0-pro")
-    response = model.predict(instances=[{"prompt": prompt}])
-    return response.predictions[0]["generated_image"]
+    response = aiplatform.TextToImageModel.from_pretrained("text-to-image").predict({"prompt": prompt})
+    return response["generated_images"][0]["image_uri"]
 
 # Función para generar texto con Vertex AI
 def generar_texto_vertex(prompt):
-    model = aiplatform.TextGeneration(model="projects/cloud-ml/models/gemini-1.0-pro")
-    response = model.predict(instances=[{"prompt": prompt}])
-    return response.predictions[0]["generated_text"]
+    response = aiplatform.TextGenerationModel.from_pretrained("text-bison").predict({"prompt": prompt})
+    return response["generated_text"]
 
 # Función para generar reflexión usando Vertex AI
 def generar_reflexion(keyword):
@@ -346,3 +344,4 @@ elif page == "Diario Reflexivo":
                 file_name="diario_reflexivo.pdf",
                 mime="application/pdf"
             )
+