|
|
import gradio as gr |
|
|
import requests |
|
|
import json |
|
|
url = "https://api.prodia.com/v1/job" |
|
|
|
|
|
headers = { |
|
|
"accept": "application/json", |
|
|
"content-type": "application/json", |
|
|
"X-Prodia-Key": "69e66898-010d-4cd1-9e22-090f73ad007b" |
|
|
} |
|
|
models = [ |
|
|
{"name": "Timeless", "url": "timeless-1.0.ckpt 1.0.ckpt [7c4971d4]"}, |
|
|
{"name": "Dreamlike-diffusion-2.0.", "url": "dreamlike-diffusion-2.0.safetensors [fdcf65e7]"}, |
|
|
{"name": "Deliberate_v2", "url": "deliberate_v2.safetensors [10ec4b29]"}, |
|
|
{"name": "Anything-v4.5-pruned", "url": "anything-v4.5-pruned.ckpt [65745d25]"}, |
|
|
] |
|
|
|
|
|
current_model = models[0] |
|
|
|
|
|
models2 = [] |
|
|
for model in models: |
|
|
model_url = f"models/{model['url']}" |
|
|
loaded_model = gr.Interface.load(model_url, live=True, preprocess=True) |
|
|
models2.append(loaded_model) |
|
|
|
|
|
|
|
|
def text_it(inputs, text_gen=text_gen): |
|
|
return text_gen(inputs) |
|
|
|
|
|
|
|
|
def set_model(current_model_index): |
|
|
global current_model |
|
|
current_model = models[current_model_index] |
|
|
return gr.update(value=f"{current_model['name']}") |
|
|
|
|
|
|
|
|
def send_it(inputs, model_choice): |
|
|
proc = models2[model_choice] |
|
|
return proc(inputs) |
|
|
|
|
|
|
|
|
with gr.Blocks() as myface: |
|
|
gr.HTML( |
|
|
|
|
|
) |
|
|
|
|
|
with gr.Row(): |
|
|
with gr.Row(): |
|
|
input_text = gr.Textbox(label="Iput Prompt", placeholder="", lines=1) |
|
|
|
|
|
model_name1 = gr.Dropdown( |
|
|
label="Choose Model", |
|
|
choices=[m["name"] for m in models], |
|
|
type="index", |
|
|
value=current_model["name"], |
|
|
interactive=True, |
|
|
) |
|
|
with gr.Row(): |
|
|
see_prompts = gr.Button("Generate Prompts") |
|
|
run = gr.Button("Generate Images", variant="primary") |
|
|
|
|
|
if __name__ == "__main__": |
|
|
demo.launch() |
|
|
|
