Qiushui567 commited on
Commit
8b2ae8a
·
verified ·
1 Parent(s): 082db9f

Create App.py

Browse files
Files changed (1) hide show
  1. App.py +29 -0
App.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ import gradio as gr
3
+ from transformers import AutoTokenizer
4
+ from optimum.onnxruntime import ORTModelForCausalLM, ORTOptions
5
+
6
+ # 选超轻量模型:微软Phi-3-mini(仅38亿参数,推理极快)
7
+ model_name = "microsoft/Phi-3-mini-4k-instruct-ONNX"
8
+
9
+ # 开启INT8量化+动态批处理,CPU计算量直接减半
10
+ options = ORTOptions(enable_int8=True, enable_dynamic_quantization=True)
11
+ model = ORTModelForCausalLM.from_pretrained(model_name, from_transformers=True, ort_options=options)
12
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
13
+
14
+ def generate_text(input_texts):
15
+     inputs = tokenizer(input_texts, return_tensors="pt", padding=True, truncation=True, max_length=32)
16
+     outputs = model.generate(
17
+         **inputs,
18
+         max_new_tokens=8,    # 生成长度砍到极致
19
+         temperature=0.1,    # 温度调低,减少随机计算
20
+         do_sample=False,
21
+         num_beams=1,
22
+         early_stopping=True # 到句号就停,不做无用功
23
+     )
24
+     return tokenizer.batch_decode(outputs, skip_special_tokens=True)
25
+
26
+ # 界面支持多行输入(批量处理请求,CPU利用率拉满)
27
+ iface = gr.Interface(fn=generate_text, inputs=gr.Textbox(multiline=True), outputs="text")
28
+ iface.launch()
29
+