Upload 3 files

README.md

@@ -0,0 +1,31 @@
+---
+title: 文本风格转换API
+emoji: 🔄
+colorFrom: blue
+colorTo: green
+sdk: gradio
+sdk_version: 3.40.0
+app_file: app.py
+pinned: false
+---
+
+# 文本风格转换API
+
+这是一个基于自定义微调模型的中文文本风格转换应用，可以将书面化、技术性文本转换为自然、口语化的表达方式。
+
+## 功能特点
+- 支持中文医学、化学等专业文本的通俗化改写
+- 实时在线转换
+- 免费API服务
+
+## API调用
+部署后可通过以下方式调用：
+```python
+import requests
+
+response = requests.post(
+    "https://yxccai-text-style-api.hf.space/api/predict",
+    json={"data": ["您要转换的文本"]}
+)
+result = response.json()
+print(result["data"][0])

app.py

@@ -0,0 +1,71 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+# 加载模型和tokenizer
+model_name = "yxccai/text-style-converter"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(
+    model_name,
+    torch_dtype=torch.float16,
+    device_map="auto"
+)
+
+def convert_text_style(input_text):
+    """文本风格转换函数"""
+    if not input_text.strip():
+        return "请输入要转换的文本"
+    
+    prompt = f"""以下是一个文本风格转换任务，请将书面化、技术性的输入文本转换为自然、口语化的表达方式。
+
+### 输入文本：
+{input_text}
+
+### 输出文本：
+"""
+    
+    inputs = tokenizer(prompt, return_tensors="pt")
+    
+    with torch.no_grad():
+        outputs = model.generate(
+            inputs.input_ids,
+            attention_mask=inputs.attention_mask,
+            max_new_tokens=500,
+            temperature=0.7,
+            do_sample=True,
+            pad_token_id=tokenizer.eos_token_id
+        )
+    
+    full_response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    # 提取生成的部分
+    if "### 输出文本：" in full_response:
+        response = full_response.split("### 输出文本：")[-1].strip()
+    else:
+        response = full_response
+    
+    return response
+
+# 创建Gradio接口
+iface = gr.Interface(
+    fn=convert_text_style,
+    inputs=gr.Textbox(
+        label="输入文本",
+        placeholder="请输入需要转换为口语化的书面文本...",
+        lines=5
+    ),
+    outputs=gr.Textbox(
+        label="输出文本",
+        lines=5
+    ),
+    title="中文文本风格转换API",
+    description="将书面化、技术性文本转换为自然、口语化表达",
+    examples=[
+        ["乙醇的检测方法包括以下几项: 1. 酸碱度检查:取20ml乙醇加20ml水,加2滴酚酞指示剂应无色。"],
+        ["本品为薄膜衣片,除去包衣后显橙红色至暗红色。"]
+    ]
+)
+
+# 启动应用
+if __name__ == "__main__":
+    iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+transformers>=4.30.0
+torch>=2.0.0
+gradio>=3.40.0
+accelerate>=0.20.0