model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "diffusers",8#     "sentence-transformers",9#     "accelerate",10#     "peft",11#     "slack-sdk",12# ]13# ///14 15try:16    import torch17    from PIL import Image18    from transformers import AutoProcessor, LightOnOCRForConditionalGeneration19    import requests20    from io import BytesIO21    22    # Load Model23    model_id = "lightonai/LightOnOCR-1B-1025"24    device = "cuda" if torch.cuda.is_available() else "cpu"25    26    processor = AutoProcessor.from_pretrained(model_id)27    model = LightOnOCRForConditionalGeneration.from_pretrained(model_id, dtype=torch.bfloat16, device_map=device, attn_implementation="sdpa")28    model.eval();29    with open('lightonai_LightOnOCR-1B-1025_0.txt', 'w', encoding='utf-8') as f:30        f.write('Everything was good in lightonai_LightOnOCR-1B-1025_0.txt')31except Exception as e:32    import os33    from slack_sdk import WebClient34    client = WebClient(token=os.environ['SLACK_TOKEN'])35    client.chat_postMessage(36        channel='#hub-model-metadata-snippets-sprint',37        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/lightonai_LightOnOCR-1B-1025_0.txt|lightonai_LightOnOCR-1B-1025_0.txt>',38    )39 40    with open('lightonai_LightOnOCR-1B-1025_0.txt', 'a', encoding='utf-8') as f:41        import traceback42        f.write('''```CODE: 43import torch44from PIL import Image45from transformers import AutoProcessor, LightOnOCRForConditionalGeneration46import requests47from io import BytesIO48 49# Load Model50model_id = "lightonai/LightOnOCR-1B-1025"51device = "cuda" if torch.cuda.is_available() else "cpu"52 53processor = AutoProcessor.from_pretrained(model_id)54model = LightOnOCRForConditionalGeneration.from_pretrained(model_id, dtype=torch.bfloat16, device_map=device, attn_implementation="sdpa")55model.eval();56```57 58ERROR: 59''')60        traceback.print_exc(file=f)61    62finally:63    from huggingface_hub import upload_file64    upload_file(65        path_or_fileobj='lightonai_LightOnOCR-1B-1025_0.txt',66        repo_id='model-metadata/code_execution_files',67        path_in_repo='lightonai_LightOnOCR-1B-1025_0.txt',68        repo_type='dataset',69    )70