Spaces:
Runtime error
Runtime error
import gradio as gr | |
import os | |
import sys | |
from pathlib import Path | |
from transformers import pipeline | |
#pipe = pipeline('text-generation', model_id='Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator') | |
models = [ | |
"Yntec/NovelAIRemix", | |
"Joeythemonster/anything-midjourney-v-4-1", | |
"prompthero/midjourney-v4-diffusion", | |
"stablediffusionapi/dreamshaper-v6", #239 | |
"stablediffusionapi/disneypixar", | |
"emilianJR/epiCRealism", | |
"prompthero/openjourney", | |
"stablediffusionapi/realistic-vision-v20-2047", | |
"stablediffusionapi/wand-magic2", | |
"dwancin/memoji", #07.11 | |
"stablediffusionapi/anime-model-v2", | |
"AIARTCHAN/MIX-Pro-V4" | |
] | |
current_model = models[0] | |
text_gen1=gr.Interface.load("models/Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator") | |
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), | |
gr.Interface.load(f"models/{models[3]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[4]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[5]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[6]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[7]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[8]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[9]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[10]}",live=True,preprocess=False), | |
gr.Interface.load(f"models/{models[11]}",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=css) as myface: | |
gr.HTML("""<!DOCTYPE html> | |
<html lang="en"> | |
<head> | |
<meta charset="utf-8" /> | |
<meta name="twitter:card" content="player"/> | |
<meta name="twitter:site" content=""/> | |
<meta name="twitter:player:width" content="100%"/> | |
<meta name="twitter:player:height" content="600"/> | |
</head> | |
</html> | |
""") | |
with gr.Row(): | |
with gr.Tab("Title"): | |
gr.HTML(""" <title>A open-beta for precious people.</title><div style="text-align: center; max-width: 1500px; margin: 0 auto;"> | |
""") | |
with gr.Tab("Description"): | |
gr.HTML("""<div style="text-align:center;"> | |
</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 automatically, by typing some words and clicking Extend Idea",lines=2) | |
see_prompts=gr.Button("Extend Idea") | |
def get_valid_prompt(text: str) -> str: | |
dot_split = text.split('.')[0] | |
n_split = text.split('\n')[0] | |
return { | |
len(dot_split) < len(n_split): dot_split, | |
len(n_split) > len(dot_split): n_split, | |
len(n_split) == len(dot_split): dot_split | |
}[True] | |
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]) | |
see_prompts.click(text_it1,inputs=[input_text],outputs=magic1) | |
#see_prompts.click(magic1=get_valid_prompt(pipe(input_text.value, max_length=77)[0]['generated_text'])) | |
myface.queue(concurrency_count=200) | |
myface.launch(inline=True, show_api=False, max_threads=400) |