plAIn / app.py
rdlf's picture
Update app.py
61a4967 verified
raw
history blame
763 Bytes
import os
from groq import Groq
import gradio as gr
client = Groq(
api_key=os.environ.get("GROQ_API_KEY"),
)
def eval_text (text):
prompt = "Eres un experto en lenguaje claro. Evalúa la calidad del lenguaje de este texto:"
input = prompt + text
chat_completion = client.chat.completions.create(
messages=[
{
"role": "user",
"content": input,
}
],
model="mixtral-8x7b-32768",
)
return (chat_completion.choices[0].message.content)
custom_image_url = "https://www.traductores.org.ar/wp-content/uploads/2021/09/CalamoCran_logo-web-.jpg"
custom_image = gr.Image(custom_image_url)
demo = gr.Interface(fn=eval_text, inputs="text", outputs="text", title="plAIn", image=custom_image)
demo.launch()