初始化项目,由ModelHub XC社区提供模型
Model: jinvbar/hebei-tourism-deepseek Source: Original Platform
This commit is contained in:
18
inference.py
Normal file
18
inference.py
Normal 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)
|
||||
Reference in New Issue
Block a user