cesar commited on
Commit
4d995a1
verified
1 Parent(s): d35d511

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -3,12 +3,6 @@ import base64
3
  import vertexai
4
  from vertexai.generative_models import GenerativeModel, Part, SafetySetting
5
 
6
- # Configuraci贸n de la clave de API de Google Cloud
7
- def configurar_api_key(api_key):
8
- """Configura la API Key para Vertex AI."""
9
- import os
10
- os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = api_key
11
-
12
  # Configuraci贸n del modelo y par谩metros globales
13
  generation_config = {
14
  "max_output_tokens": 8192,
@@ -35,14 +29,19 @@ safety_settings = [
35
  ),
36
  ]
37
 
 
 
 
 
 
 
 
 
38
  def revisar_examen(api_key, pdf_docente, pdf_alumno):
39
  """Funci贸n principal para comparar los PDFs del docente y del alumno."""
40
  try:
41
- # Configurar la API Key de Google Cloud
42
- configurar_api_key(api_key)
43
-
44
- # Inicializar Vertex AI
45
- vertexai.init(project="deploygpt", location="us-central1")
46
 
47
  # Leer los datos de los PDFs desde la ruta proporcionada por Gradio
48
  with open(pdf_docente.name, "rb") as docente_file:
 
3
  import vertexai
4
  from vertexai.generative_models import GenerativeModel, Part, SafetySetting
5
 
 
 
 
 
 
 
6
  # Configuraci贸n del modelo y par谩metros globales
7
  generation_config = {
8
  "max_output_tokens": 8192,
 
29
  ),
30
  ]
31
 
32
+ def configurar_vertex_ai(api_key):
33
+ """Configura Vertex AI para usar una API Key de Google Cloud."""
34
+ vertexai.init(
35
+ project="deploygpt",
36
+ location="us-central1",
37
+ credentials={"api_key": api_key}
38
+ )
39
+
40
  def revisar_examen(api_key, pdf_docente, pdf_alumno):
41
  """Funci贸n principal para comparar los PDFs del docente y del alumno."""
42
  try:
43
+ # Configurar Vertex AI con la API Key proporcionada
44
+ configurar_vertex_ai(api_key)
 
 
 
45
 
46
  # Leer los datos de los PDFs desde la ruta proporcionada por Gradio
47
  with open(pdf_docente.name, "rb") as docente_file: