app.py

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "jinaai/reader-lm-1.5b"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto").to("cuda")

print("模型加载成功！")