locorene1000 commited on
Commit
649a73e
verified
1 Parent(s): d3e6334
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,7 +53,7 @@ def mejorar_resolucion(input_text):
53
  prompt = f"{instrucciones}\n\nTexto original a mejorar o pregunta:\n{input_text}\n\nVersi贸n mejorada o respuesta:"
54
 
55
  # Crear el pipeline de generaci贸n de texto
56
- pipe = pipeline("text-generation", model="mistralai/Mistral-Nemo-Instruct-2407")
57
 
58
  # Generar respuesta
59
  response = pipe(prompt, max_length=512, num_return_sequences=1, temperature=0.7)
@@ -62,7 +62,7 @@ def mejorar_resolucion(input_text):
62
  # Definici贸n de la interfaz de Gradio
63
  with gr.Blocks() as demo:
64
  gr.Markdown("# Mejora de Resoluciones Judiciales con GPT Civil")
65
- gr.Markdown("Utiliza el modelo Mistral-Nemo-Instruct-2407 para mejorar borradores de resoluciones judiciales.")
66
  input_text = gr.Textbox(label="Introduce tu resoluci贸n judicial o pregunta")
67
  output_text = gr.Textbox(label="Resoluci贸n mejorada o respuesta")
68
  submit_button = gr.Button("Mejorar")
 
53
  prompt = f"{instrucciones}\n\nTexto original a mejorar o pregunta:\n{input_text}\n\nVersi贸n mejorada o respuesta:"
54
 
55
  # Crear el pipeline de generaci贸n de texto
56
+ pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-8B-Instruct")
57
 
58
  # Generar respuesta
59
  response = pipe(prompt, max_length=512, num_return_sequences=1, temperature=0.7)
 
62
  # Definici贸n de la interfaz de Gradio
63
  with gr.Blocks() as demo:
64
  gr.Markdown("# Mejora de Resoluciones Judiciales con GPT Civil")
65
+ gr.Markdown("Utiliza el modelo Meta-Llama-3.1-8B-Instruct para mejorar borradores de resoluciones judiciales.")
66
  input_text = gr.Textbox(label="Introduce tu resoluci贸n judicial o pregunta")
67
  output_text = gr.Textbox(label="Resoluci贸n mejorada o respuesta")
68
  submit_button = gr.Button("Mejorar")