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    import os15    from huggingface_hub import InferenceClient16    17    client = InferenceClient(18        provider="auto",19        api_key=os.environ["HF_TOKEN"],20    )21    22    with open("cat.png", "rb") as image_file:23       input_image = image_file.read()24    25    # output is a PIL.Image object26    image = client.image_to_image(27        input_image,28        prompt="Turn the cat into a tiger.",29        model="Qwen/Qwen-Image-Edit",30    )31    with open('Qwen_Qwen-Image-Edit_2.txt', 'w', encoding='utf-8') as f:32        f.write('Everything was good in Qwen_Qwen-Image-Edit_2.txt')33except Exception as e:34    import os35    from slack_sdk import WebClient36    client = WebClient(token=os.environ['SLACK_TOKEN'])37    client.chat_postMessage(38        channel='#exp-slack-alerts',39        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/Qwen_Qwen-Image-Edit_2.txt|Qwen_Qwen-Image-Edit_2.txt>',40    )41 42    with open('Qwen_Qwen-Image-Edit_2.txt', 'a', encoding='utf-8') as f:43        import traceback44        f.write('''```CODE: 45import os46from huggingface_hub import InferenceClient47 48client = InferenceClient(49    provider="auto",50    api_key=os.environ["HF_TOKEN"],51)52 53with open("cat.png", "rb") as image_file:54   input_image = image_file.read()55 56# output is a PIL.Image object57image = client.image_to_image(58    input_image,59    prompt="Turn the cat into a tiger.",60    model="Qwen/Qwen-Image-Edit",61)62```63 64ERROR: 65''')66        traceback.print_exc(file=f)67    68finally:69    from huggingface_hub import upload_file70    upload_file(71        path_or_fileobj='Qwen_Qwen-Image-Edit_2.txt',72        repo_id='model-metadata/code_execution_files',73        path_in_repo='Qwen_Qwen-Image-Edit_2.txt',74        repo_type='dataset',75    )76