jinv2 commited on
Commit
b0f5b88
·
verified ·
1 Parent(s): b4b0a92

Create app.py

Browse files

Add app.py and requirements.txt

Files changed (1) hide show
  1. app.py +23 -0
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ generator = pipeline("text-generation", model="jinv2/finetuned-zhengding-tourism", device=-1)
5
+
6
+ def predict(input_text):
7
+ if not input_text.endswith('\n'):
8
+ input_text += '\n'
9
+ output = generator(
10
+ f"用户: {input_text}",
11
+ max_length=200,
12
+ num_return_sequences=1,
13
+ do_sample=True,
14
+ temperature=0.7,
15
+ top_k=50,
16
+ top_p=0.95,
17
+ pad_token_id=generator.tokenizer.pad_token_id,
18
+ eos_token_id=generator.tokenizer.eos_token_id
19
+ )[0]["generated_text"]
20
+ return output
21
+
22
+ interface = gr.Interface(fn=predict, inputs="text", outputs="text", title="正定旅游助手")
23
+ interface.launch(server_name="0.0.0.0", server_port=7860)