File size: 2,844 Bytes
8622a65 5b69b8e 8622a65 8ab6a28 5b69b8e 8622a65 33c9f94 8ab6a28 f0ceea8 8ab6a28 5b69b8e 8622a65 5b69b8e 6bef07c 2d6695a 6bef07c 99628be 8622a65 5b69b8e 8622a65 5b69b8e 8622a65 9331b8e d382edc 0198d09 5b69b8e 8622a65 5b69b8e 8622a65 5b69b8e 8ab6a28 8622a65 5b69b8e 8622a65 a90e46c 8622a65 a90e46c 11c4d5c 85db0a7 2f20bf3 8622a65 ccedfec 73a8264 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
# Importar las bibliotecas necesarias
from huggingface_hub import InferenceClient
import gradio as gr
# Crear un cliente de inferencia para el modelo preentrenado Mixtral-8x7B-Instruct-v0.1
client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
# Función para formatear el prompt con historial
def format_prompt(message, history, system_prompt):
prompt = "<s>"
for user_prompt, bot_response in history:
prompt += f"[INST] {user_prompt} [/INST]"
prompt += f" {bot_response}</s> "
prompt += f"[INST] {system_prompt}, {message} [/INST]"
return prompt
# Función para generar respuestas dada una serie de parámetros
def generate(
prompt,
history,
system_prompt= "Experto en servicios de abastecimiento, depuración, reutilización, alcantarillado y calidad del agua, para la empresa Canal de Isabel II.",
temperature=0.5,
max_new_tokens=4096,
top_p=0.95,
repetition_penalty=1.0,
input_max_length=256,):
# Ajustar valores de temperatura y top_p para asegurar que estén en el rango adecuado
temperature = float(temperature)
if temperature < 1e-2:
temperature = 1e-2
top_p = float(top_p)
# Configurar los parámetros para la generación de texto
generate_kwargs = dict(
temperature=temperature,
max_new_tokens=max_new_tokens,
top_p=top_p,
repetition_penalty=repetition_penalty,
do_sample=True,
seed=42,
)
# Formatear el prompt y obtener la respuesta del modelo de manera continua
formatted_prompt = format_prompt(prompt, history, system_prompt)
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
output = ""
# Iterar a través de las respuestas en el stream
for response in stream:
output += response.token.text
yield output
return output
# Crear una interfaz de chat Gradio con el modelo generativo
gr.ChatInterface(
fn=generate,
chatbot=gr.Chatbot(
avatar_images=["./botm.png", "./logoCanal.png"],
bubble_full_width=False,
show_label=False,
show_share_button=False,
show_copy_button=True,
likeable=True,
layout="panel",
height=500,
),
textbox=gr.Textbox(placeholder="¿Qué es una tubería de polietiteno?", container=False, scale=7),
theme="soft",
title="Área Tajo: Experto en sector del agua del Canal de Isabel ll",
description='Autor: Antonio Fernández de <a href=\"https://www.canaldeisabelsegunda.es/\">Canal de Isabel II</a>. Aplicación desarrollada para el Área Conservación Sistema Tajo',
retry_btn="Repetir",
undo_btn="Deshacer",
clear_btn="Borrar",
submit_btn="Enviar",
concurrency_limit=20,
).launch(show_api=False)
|