model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "accelerate",8#     "peft",9#     "slack-sdk",10# ]11# ///12 13try:14    from diffusers import DiffusionPipeline15    from diffusers.utils import load_image16    17    pipe = DiffusionPipeline.from_pretrained("dx8152/Fusion_lora")18    19    prompt = "Turn this cat into a dog"20    input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")21    22    image = pipe(image=input_image, prompt=prompt).images[0]23    with open('dx8152_Fusion_lora_2.txt', 'w', encoding='utf-8') as f:24        f.write('Everything was good in dx8152_Fusion_lora_2.txt')25except Exception as e:26    import os27    from slack_sdk import WebClient28    client = WebClient(token=os.environ['SLACK_TOKEN'])29    client.chat_postMessage(30        channel='#exp-slack-alerts',31        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/dx8152_Fusion_lora_2.txt|dx8152_Fusion_lora_2.txt>',32    )33 34    with open('dx8152_Fusion_lora_2.txt', 'a', encoding='utf-8') as f:35        import traceback36        f.write('''```CODE: 37from diffusers import DiffusionPipeline38from diffusers.utils import load_image39 40pipe = DiffusionPipeline.from_pretrained("dx8152/Fusion_lora")41 42prompt = "Turn this cat into a dog"43input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")44 45image = pipe(image=input_image, prompt=prompt).images[0]46```47 48ERROR: 49''')50        traceback.print_exc(file=f)51    52finally:53    from huggingface_hub import upload_file54    upload_file(55        path_or_fileobj='dx8152_Fusion_lora_2.txt',56        repo_id='model-metadata/code_execution_files',57        path_in_repo='dx8152_Fusion_lora_2.txt',58        repo_type='dataset',59    )60