Upload 3 files

pulse_model_deployment/README.md

@@ -0,0 +1,10 @@
+
+# PULSE 疾病问答系统
+
+## 介绍
+这是一个基于PULSE模型的疾病问答系统，用户可以输入症状描述，系统将生成对应的疾病回答。
+
+## 如何使用
+1. 通过Gradio界面与模型进行交互。
+2. 在输入框中输入症状描述，点击“提交”按钮即可获取模型的回答。
+    

pulse_model_deployment/app.py

@@ -0,0 +1,20 @@
+
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+# 加载模型
+model_name = "OpenMEDLab/PULSE-7bv5"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# 定义Gradio接口
+def predict(symptoms):
+    inputs = tokenizer(symptoms, return_tensors="pt")
+    outputs = model.generate(**inputs, max_length=200)
+    result_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return result_text
+
+# 创建Gradio界面
+iface = gr.Interface(fn=predict, inputs="text", outputs="text", title="PULSE 疾病问答系统", description="输入症状描述获取模型的回答")
+iface.launch()
+    

pulse_model_deployment/requirements.txt

@@ -0,0 +1,5 @@
+
+gradio
+transformers
+torch
+