Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 42 |
-
|
| 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:
|