File size: 3,165 Bytes
6adca2a 21e5a14 a0f0274 6adca2a a0f0274 6adca2a c5a6204 6adca2a 356a74a bcc0f80 356a74a 6adca2a 50ab07e 6adca2a 356a74a 6adca2a 631fee3 6adca2a 223f7f1 d1c3924 6adca2a d2243dd a0f0274 709aafc a0f0274 d6787cb a0f0274 c66f766 f2aba62 6adca2a a0f0274 2972058 f2aba62 a0f0274 50ab07e 6adca2a c11029c 6adca2a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
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"""
<p style='text-align: center;font-size: large'>Xmodel + Magic Prompt </p>
"""
article = r"""
<br><br><br><br><br><br><br><br><br><br><br><br>
"""
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"""<title>Как использовать</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;">
<h4>1. </h4>
<h4>2. </h4>
<h4>3. </h4>
<h4>4. </h4>
<h4>5. </h4></div>
""")
with gr.Row():
with gr.Column(scale=100):
magic1=gr.Textbox(lines=4)
gr.HTML("""<style> .gr-button {
color: white !important;
border-color: #000000 !important;
background: #499afc !important;
}</style>""")
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) |