| 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() |