# 下载套件源码
from openmind import pipeline,is_torch_npu_available
import argparse
def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
    "--model_name_or_path",
    type=str,
    help="Path to model",
    default=None)
    args = parser.parse_args()
    return args  
args = parse_args()
if is_torch_npu_available(): 
    device = "npu:0" 
else: 
    device = "cpu"
pipe = pipeline("text-classification", model=args.model_name_or_path, framework="pt",device=device)
output = pipe("Hello I'm Omar and I live in Zürich.")
print(output)