Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,108 Bytes
8110449 04ef268 e5450bc 04ef268 |
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 |
import spaces
import gradio as gr
from src.util.base import *
from src.util.params import *
@spaces.GPU()
def display_negative_images(
prompt, seed, num_inference_steps, negative_prompt="", progress=gr.Progress()
):
text_embeddings = get_text_embeddings(prompt)
text_embeddings_neg = get_text_embeddings(prompt, negative_prompt=negative_prompt)
latents = generate_latents(seed)
progress(0)
images = generate_images(latents, text_embeddings, num_inference_steps)
progress(0.5)
images_neg = generate_images(latents, text_embeddings_neg, num_inference_steps)
fname = "negative"
tab_config = {
"Tab": "Negative",
"Prompt": prompt,
"Negative Prompt": negative_prompt,
"Number of Inference Steps per Image": num_inference_steps,
"Seed": seed,
}
imgs_list = []
imgs_list.append((images, "Without Negative Prompt"))
imgs_list.append((images_neg, "With Negative Prompt"))
export_as_zip(imgs_list, fname, tab_config)
return images, images_neg, f"outputs/{fname}.zip"
__all__ = ["display_negative_images"]
|