Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import gradio as gr | |
# Cargar un modelo de lenguaje médico de Hugging Face | |
modelo_medico = pipeline("text-generation", model="microsoft/BioGPT-Large") | |
# Función para hacer preguntas al bot | |
def pregunta_medica(pregunta): | |
respuesta = modelo_medico( | |
f"Eres un experto médico con vasto conocimiento en fisiología, bioquímica, farmacología y otras áreas médicas. Responde de manera sarcástica y directa, como el Dr. House. Pregunta: {pregunta}", | |
max_length=150 | |
) | |
return respuesta[0]['generated_text'] | |
# Crear la interfaz web con Gradio | |
interfaz = gr.Interface( | |
fn=pregunta_medica, | |
inputs="text", | |
outputs="text", | |
title="Dr. House Bot", | |
description="Pregúntale al Dr. House sobre medicina." | |
) | |
# Lanzar la interfaz | |
interfaz.launch() |