jinv2 commited on
Commit
d35c471
·
1 Parent(s): 3d60082

Fix syntax error in app.py by removing invalid characters

Browse files
Files changed (1) hide show
  1. app.py +29 -29
app.py CHANGED
@@ -1,35 +1,35 @@
1
- 进口 市井 as 格拉格
2
- 谢电压器 进口 AutoModelForCausalLM, 自动代币器
3
- 炎特 进口 鎷岀特濡垝
4
 
5
- # 椹蟽
6
- 垝_ 转发 = "jinv2/ai-求婚者-航海家-濡垝"
7
- 地墿垝号 = AutoModelForCausalLM.对此进行了 假发。("distilgpt2")
8
- 领牌 = 自动代币器.对此进行了 假发。(垝_ 转发)
9
- 濡垝 = 鎷岀特濡垝.对此进行了 传单。(地墿垝号, 垝_ 转发)
10
 
11
- #定义生成??数
12
- ⁇ , 生成_建设(提款):
13
- ,进入 = 领牌(提款, return_tensors="pt")
14
- 出来 = 濡垝.生成(
15
- **,进入,
16
- 最大的长度=200,
17
- do_sample=真的,
18
- 温西斯的=0.7,
19
- top_k=50绔 涔诲芥澶澧绔 涔诲芥澶澧 绔 涔诲芥澶澧,
20
- 顶部_p=0.9,
21
- num_return_oetal=1
22
  )
23
- 武峰 = 领牌.解码(出来[0], 鱼过_特 000浜垮厓锛屽悓姣斿??_代币=真的)
24
- 啊回武峰
25
 
26
- # 2 建设世界面
27
- 扣口 = 格拉格.扣口(
28
- fn=生成_建设,
29
- ,进入=格拉格.文本 000浜垮厓锛屽悓姣斿??(x),在2019年10月10日开始实施,在2019年10月10日开始实施,在2019年10月10日开始实施,在2019年10月10日开始实施衣叶="进入提款", 联盟合体=根福最新的AI行学问,2025年年学问学问:),
30
- 出来=格拉格.文本 000浜垮厓锛屽悓姣斿??(x),在2019年10月10日开始实施,在2019年10月10日开始实施,在2019年10月10日开始实施,在2019年10月10日开始实施衣叶=的「生成为€」),
31
- 主题=2025年 年航 ⁇ 的求者,
32
- 灏="鐜弬进姘戝竞2025年AI行业的职业建设鏂()鏅微旱distilgpt2濡垝锛商行:鏂股锛商行数字有限公司,生成财果可以不理智
33
  )
34
 
35
- 扣口.推荐()
 
1
+ import gradio as gr
2
+ from transformers import AutoModelForCausalLM, AutoTokenizer
3
+ from peft import PeftModel
4
 
5
+ # 加载模型
6
+ MODEL_REPO = "jinv2/ai-job-navigator-model"
7
+ base_model = AutoModelForCausalLM.from_pretrained("distilgpt2")
8
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO)
9
+ model = PeftModel.from_pretrained(base_model, MODEL_REPO)
10
 
11
+ # 定义生成函数
12
+ def generate_advice(prompt):
13
+ inputs = tokenizer(prompt, return_tensors="pt")
14
+ outputs = model.generate(
15
+ **inputs,
16
+ max_length=200,
17
+ do_sample=True,
18
+ temperature=0.7,
19
+ top_k=50,
20
+ top_p=0.9,
21
+ num_return_sequences=1
22
  )
23
+ response = tokenizer.decode(outputs[0], skip_special_tokens=True)
24
+ return response
25
 
26
+ # 创建 Gradio 界面
27
+ interface = gr.Interface(
28
+ fn=generate_advice,
29
+ inputs=gr.Textbox(label="输入提示", placeholder="根据最新的AI行业趋势,提供2025年的职业建议:"),
30
+ outputs=gr.Textbox(label="生成结果"),
31
+ title="AI Job Navigator 2025",
32
+ description="输入提示以获取 2025 AI 行业的职业建议(基于微调的 distilgpt2 模型)。注意:由于训练数据有限,生成结果可能不理想。"
33
  )
34
 
35
+ interface.launch()