import gradio as gr import os import sys from pathlib import Path models = [ "Yntec/Lyriel", "Yntec/3DKX", "Yntec/Dreamshaper8", ] current_model = models[0] text_gen1=gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link") models2=[ gr.Interface.load(f"models/{models[0]}",live=True,preprocess=False), gr.Interface.load(f"models/{models[1]}",live=True,preprocess=False), gr.Interface.load(f"models/{models[2]}",live=True,preprocess=False), ] def text_it1(inputs,text_gen1=text_gen1): go_t1=text_gen1(inputs) return(go_t1) def set_model(current_model): current_model = models[current_model] return gr.update(label=(f"{current_model}")) def send_it1(inputs, model_choice): proc1=models2[model_choice] output1=proc1(inputs) return(output1) css="""""" title = r"""

Xmodel + Magic Prompt

""" article = r"""











""" with gr.Blocks(css=css) as myface: # Добавление title gr.HTML(title) with gr.Row(): with gr.Tab("Рисовать"): gr.HTML(""" """) with gr.Tab("Как использовать"): gr.HTML(f"""Как использовать

1.

2.

3.

4.

5.

""") with gr.Row(): with gr.Column(scale=100): magic1=gr.Textbox(lines=4) gr.HTML("""""") run=gr.Button("Создать изображение") with gr.Row(): with gr.Column(scale=100): #Model selection dropdown model_name1 = gr.Dropdown(label="Модель", choices=[m for m in models], type="index", value=current_model, interactive=True) with gr.Row(): with gr.Column(scale=50): input_text=gr.Textbox(label="Идея изображения:",lines=2) use_short=gr.Button("Активировать идею") see_prompts=gr.Button("Генерировать Magic промт") with gr.Row(): with gr.Column(style="width=800px"): output1=gr.Image(label=(f"{current_model}")) # Добавление article gr.HTML(article) def short_prompt(inputs): return(inputs) model_name1.change(set_model,inputs=model_name1,outputs=[output1]) run.click(send_it1, inputs=[magic1, model_name1], outputs=[output1]) use_short.click(short_prompt,inputs=[input_text],outputs=magic1) see_prompts.click(text_it1,inputs=[input_text],outputs=magic1) myface.queue(concurrency_count=200) myface.launch(inline=True, show_api=False, max_threads=400)