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", "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", "goofyai/3d_render_style_xl" ] 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), ] 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(""" """) with gr.Row(): with gr.Tab("Title"): gr.HTML(""" A open-beta for precious people.
""") with gr.Tab("Description"): gr.HTML("""
""") with gr.Row(): with gr.Column(scale=100): magic1 = gr.Textbox(lines=4) gr.HTML("""""") 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)