File size: 615 Bytes
c6ad148 6e6ab5c c6ad148 6e6ab5c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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() |