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:
|