import gradio as gr
from transformers import pipeline

# 加载 Hugging Face 预训练模型
model_name = "YLX1965/medical-model"  # 你的模型名称
pipe = pipeline("text-generation", model=model_name)

# 定义 Gradio 交互函数
def generate_text(prompt):
    result = pipe(prompt, max_length=200)[0]["generated_text"]
    return result

# 创建 Gradio 界面
interface = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Medical Text Generation",
    description="输入医疗相关的文本，模型将自动补全内容。",
)

# 运行 Gradio 应用
interface.launch()