model-metadata/custom-code-py-files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "transformers",5#     "torch",6# ]7# ///8 9try:10    # Use a pipeline as a high-level helper11    from transformers import pipeline12    13    pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)14    messages = [15        {"role": "user", "content": "Who are you?"},16    ]17    pipe(messages)18except Exception as e:19    exception_file = 'deepseek-ai_DeepSeek-R1_0_exception.txt'20    with open(exception_file, 'w') as f:21        import traceback22        traceback.print_exc(file=f)23    # Upload exception log to HuggingFace24    from huggingface_hub import upload_file25    upload_file(26        path_or_fileobj=exception_file,27        repo_id='model-metadata/model-code-exception',28        path_in_repo='09-07-25/deepseek-ai_DeepSeek-R1_0_exception.txt',29        repo_type='dataset',30    )