ToyWorld / app.py
Yntec's picture
Update app.py
1c3bb07
raw
history blame
No virus
3.56 kB
import gradio as gr
import os
import sys
from pathlib import Path
import random
import string
import time
from queue import Queue
from threading import Thread
import emoji
text_gen=gr.Interface.load("spaces/phenomenon1981/MagicPrompt-Stable-Diffusion")
def get_prompts(prompt_text):
if prompt_text:
return text_gen(prompt_text + " Dream")
else:
return text_gen("")
models = [
"Yntec/OpenLexica",
"Yntec/MapleSyrup",
]
current_model = models[0]
models2=[
gr.Interface.load(f"models/{models[0]}",live=True,preprocess=False),
gr.Interface.load(f"models/{models[1]}",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=""""""
with gr.Blocks(css='style.css') as demo:
gr.HTML(
"""
<div style="text-align: center; max-width: 1200px; margin: 0 auto;">
<div>
<style>
h1 {
font-size: 6em;
color: #ffffff;
margin-top: 30px;
margin-bottom: 30px;
text-shadow: 3px 3px 0 rgba(0, 0, 0, 1) !important;
}
</style>
<body>
<div class="center"><h1>ToyWorld XL 401</h1>
</div>
</body>
</div>
<p style="margin-bottom: 10px; color: #ffffff;">
PlaceHolder
<a href="https://twitter.com/DavidJohnstonxx/">original space created by Phenomenon1981</a>.
</p>
<p style="margin-bottom: 10px; color: #ffaa66; font-size: 98%">
❤️ Press the Like Button if you enjoy my space! ❤️</a>
</p>
</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: #006699 !important;
}</style>""")
run=gr.Button("Generate Image")
with gr.Row():
with gr.Column(scale=100):
#Model selection dropdown
model_name1 = gr.Dropdown(label="Select Model", choices=[m for m in models], type="index", value=current_model, interactive=True)
with gr.Row():
with gr.Column(style="width=800px"):
output1=gr.Image(label=(f"{current_model}"))
with gr.Row():
with gr.Column(scale=50):
input_text=gr.Textbox(label="Use this box to extend an idea automagically, by typing some words and clicking Extend Idea",lines=2)
use_short=gr.Button("Use Short Prompt")
see_prompts=gr.Button("Extend Idea")
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)