model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "numpy",5#     "einops",6#     "pandas",7#     "protobuf",8#     "torch",9#     "torchvision",10#     "transformers",11#     "timm",12#     "diffusers",13#     "sentence-transformers",14#     "accelerate",15#     "peft",16#     "slack-sdk",17# ]18# ///19 20try:21    from diffusers import DiffusionPipeline22    23    pipe = DiffusionPipeline.from_pretrained("inclusionAI/Ming-flash-omni-Preview")24    25    prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"26    image = pipe(prompt).images[0]27    with open('inclusionAI_Ming-flash-omni-Preview_0.txt', 'w', encoding='utf-8') as f:28        f.write('Everything was good in inclusionAI_Ming-flash-omni-Preview_0.txt')29except Exception as e:30    import os31    from slack_sdk import WebClient32    client = WebClient(token=os.environ['SLACK_TOKEN'])33    client.chat_postMessage(34        channel='#hub-model-metadata-snippets-sprint',35        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/inclusionAI_Ming-flash-omni-Preview_0.txt|inclusionAI_Ming-flash-omni-Preview_0.txt>',36    )37 38    with open('inclusionAI_Ming-flash-omni-Preview_0.txt', 'a', encoding='utf-8') as f:39        import traceback40        f.write('''```CODE: 41from diffusers import DiffusionPipeline42 43pipe = DiffusionPipeline.from_pretrained("inclusionAI/Ming-flash-omni-Preview")44 45prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"46image = pipe(prompt).images[0]47```48 49ERROR: 50''')51        traceback.print_exc(file=f)52    53finally:54    from huggingface_hub import upload_file55    upload_file(56        path_or_fileobj='inclusionAI_Ming-flash-omni-Preview_0.txt',57        repo_id='model-metadata/code_execution_files',58        path_in_repo='inclusionAI_Ming-flash-omni-Preview_0.txt',59        repo_type='dataset',60    )61