rdlf's picture
Update app.py
674d338 verified
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="somosnlp/Phi-2-LenguajeClaro", trust_remote_code=True)
import gradio as gr
mis_ejemplos = ['''El crecimiento económico, que el año pasado fue cinco veces superior a la media de la zona euro,
y la mejora del empleo con cifras récord de afiliación, permiten mejorar las previsiones de déficit sin aplicar recortes
y reforzando el Estado de Bienestar.
El inicio de 2024 continúa con la senda de descenso y el déficit del Estado hasta febrero se sitúa en el 0,61%''']
def eval_text (text):
new_prompt = f"""###System: Lee el siguiente texto y hazlo más claro:
###Texto:
{text}
###Texto aclarado:
"""
result = pipe(new_prompt, max_length=2000)
return (result[0]['generated_text'])
demo = gr.Interface(fn=eval_text, inputs="text", outputs="text", title="Lenguaje Claro", description="modelo de lenguaje entrenado para aclarar el lenguaje de la administración pública española", examples=mis_ejemplos)
demo.launch()