AlvearRendon / app.py
AlbertDuvan's picture
Upload 2 files
dd5582f verified
raw
history blame
843 Bytes
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()