A

app.py

@@ -1,16 +1,9 @@
 import requests
 import json
 import gradio as gr
-import spaces
 
-# Configuración de la API de NVIDIA
-base_url = "https://integrate.api.nvidia.com/v1"
-api_key = "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"
-headers = {
-    "Authorization": f"Bearer {api_key}",
-    "Content-Type": "application/json",
-    "Accept": "application/json"
-}
+# Configuración de la API de Hugging Face
+base_url = "https://locorene1000-llama3-1.hf.space/call/mejorar_resolucion"
 
 # Instrucciones para el modelo
 instrucciones = """
@@ -36,31 +29,41 @@ Revisas el documento para asegurar:
 - Revisión integral y presentación concisa y unificada de la información.
 """
 
-@spaces.GPU(duration=120)
-def mejorar_resolucion(input_text):
-    # Construcción del prompt con instrucciones unificadas
-    prompt = f"{instrucciones}\n{instrucciones_legales}\nTexto o pregunta:\n{input_text}\nRespuesta:"
-    
+def iniciar_consulta(input_text):
     payload = {
-        "model": "nvidia/nemotron-4-340b-instruct",
-        "messages": [{"role": "user", "content": prompt}],
-        "temperature": 0.2,
-        "top_p": 0.7,
-        "max_tokens": 1024,
-        "stream": False
+        "data": [input_text]
+    }
+    headers = {
+        "Content-Type": "application/json"
     }
+    
+    response = requests.post(base_url, headers=headers, data=json.dumps(payload))
+    
+    if response.status_code == 200:
+        event_id = response.json().get("event_id")
+        return event_id
+    else:
+        return f"Error {response.status_code}: {response.text}"
 
-    response = requests.post(f"{base_url}/chat/completions", headers=headers, data=json.dumps(payload))
+def obtener_resultado(event_id):
+    response = requests.get(f"{base_url}/{event_id}", headers={"Accept": "text/event-stream"})
     
     if response.status_code == 200:
-        response_body = response.json()
-        return response_body['choices'][0]['message']['content']
+        return response.text
     else:
         return f"Error {response.status_code}: {response.text}"
 
+def mejorar_resolucion(input_text):
+    event_id = iniciar_consulta(input_text)
+    if "Error" in event_id:
+        return event_id
+    
+    resultado = obtener_resultado(event_id)
+    return resultado
+
 # Definición de la interfaz de Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("# Mejora de Resoluciones Judiciales y Respuestas con Nemotron")
+    gr.Markdown("# Mejora de Resoluciones Judiciales y Respuestas con Ministro en Visita")
     input_text = gr.Textbox(label="Introduce tu resolución judicial o pregunta")
     output_text = gr.Textbox(label="Respuesta mejorada o respuesta corta")
     submit_button = gr.Button("Enviar")