| import gradio as gr | |
| from multit2i import ( | |
| load_models, | |
| find_model_list, | |
| infer_multi, | |
| save_gallery_images, | |
| change_model, | |
| get_model_info_md, | |
| loaded_models, | |
| ) | |
| models = find_model_list("John6666", ["anime"], "", "downloads", 40) | |
| load_models(models, 10) | |
| css = """""" | |
| with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", css=css) as demo: | |
| with gr.Column(): | |
| model_name = gr.Dropdown(label="Select Model", choices=list(loaded_models.keys()), value=list(loaded_models.keys())[0], allow_custom_value=True) | |
| model_info = gr.Markdown(value=get_model_info_md(list(loaded_models.keys())[0])) | |
| image_num = gr.Slider(label="Number of Images", minimum=1, maximum=8, value=1, step=1) | |
| recom_prompt = gr.Checkbox(label="Recommended Prompt", value=True) | |
| prompt = gr.Text(label="Prompt", lines=1, max_lines=8, placeholder="1girl, solo, ...") | |
| run_button = gr.Button("Generate Image") | |
| results = gr.Gallery(label="Gallery", interactive=False, show_download_button=True, show_share_button=False, | |
| container=True, format="png", object_fit="contain") | |
| image_files = gr.Files(label="Download", interactive=False) | |
| clear_results = gr.Button("Clear Gallery and Download") | |
| gr.Markdown( | |
| f"""This demo was created in reference to the following demos. | |
| - [Nymbo/Flood](https://huggingface.co/spaces/Nymbo/Flood). | |
| - [Yntec/ToyWorldXL](https://huggingface.co/spaces/Yntec/ToyWorldXL). | |
| """ | |
| ) | |
| gr.DuplicateButton(value="Duplicate Space") | |
| model_name.change(change_model, [model_name], [model_info], queue=False, show_api=False) | |
| gr.on( | |
| triggers=[run_button.click, prompt.submit], | |
| fn=infer_multi, | |
| inputs=[prompt, model_name, recom_prompt, image_num, results], | |
| outputs=[results], | |
| queue=True, | |
| show_progress="full", | |
| show_api=True, | |
| ).success(save_gallery_images, [results], [results, image_files], queue=False, show_api=False) | |
| clear_results.click(lambda: (None, None), None, [results, image_files], queue=False, show_api=False) | |
| demo.queue() | |
| demo.launch() | |