AffinityConnect/LiteLLM

1import gradio as gr2from transformers import pipeline3 4# Cargar el modelo (usamos un modelo gratuito de Hugging Face)5chat = pipeline("conversational", model="mistralai/Mistral-7B-Instruct-v0.2")6 7# Función para la conversación8def chatbot(input_text):9    return chat(input_text)[0]["generated_text"]10 11# Crear la interfaz Gradio12interface = gr.Interface(fn=chatbot, inputs="text", outputs="text")13interface.launch()14