初始化项目,由ModelHub XC社区提供模型

Model: jinvbar/hebei-tourism-deepseek
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-08-06 03:44:13 +08:00
commit 0a2e69ec22
15 changed files with 159994 additions and 0 deletions

18
inference.py Normal file
View File

@@ -0,0 +1,18 @@
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
# 加载模型和分词器
model_path = "path/to/your/model" # ModelScope 会自动设置路径
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(model_path)
def inference(input_text):
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs, max_length=50)
return tokenizer.decode(outputs[0], skip_special_tokens=True)
# 示例输入
if __name__ == "__main__":
input_text = "河北旅游推荐"
result = inference(input_text)
print(result)