Clean-up code.

app.py

@@ -13,13 +13,13 @@ import torch
 from diffusers import DiffusionPipeline
 from typing import Tuple
 
-#Check for the Model Base..//
+# Check for the Model Base..//
 
 
+bad_words = json.loads(os.getenv("BAD_WORDS", "[]"))
+bad_words_negative = json.loads(os.getenv("BAD_WORDS_NEGATIVE", "[]"))
+default_negative = os.getenv("default_negative", "")
 
-bad_words = json.loads(os.getenv('BAD_WORDS', "[]"))
-bad_words_negative = json.loads(os.getenv('BAD_WORDS_NEGATIVE', "[]"))
-default_negative = os.getenv("default_negative","")
 
 def check_text(prompt, negative=""):
     for i in bad_words:
@@ -31,27 +31,22 @@ def check_text(prompt, negative=""):
     return False
 
 
-
 style_list = [
-
     {
         "name": "2560 x 1440",
         "prompt": "hyper-realistic 4K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
         "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
     },
-
     {
         "name": "Photo",
         "prompt": "cinematic photo {prompt}. 35mm photograph, film, bokeh, professional, 4k, highly detailed",
         "negative_prompt": "drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly",
-    },   
-
+    },
     {
         "name": "Cinematic",
         "prompt": "cinematic still {prompt}. emotional, harmonious, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy",
         "negative_prompt": "anime, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured",
     },
-
     {
         "name": "Anime",
         "prompt": "anime artwork {prompt}. anime style, key visual, vibrant, studio anime, highly detailed",
@@ -73,6 +68,7 @@ styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "2560 x 1440"
 
+
 def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     if not negative:
@@ -80,18 +76,12 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
     return p.replace("{prompt}", positive), n + negative
 
 
-
-    
-
 DESCRIPTION = """## MidJourney
 
 Drop your best results in the community: [rb.gy/klkbs7](http://rb.gy/klkbs7), Have you tried the dalle collage space? [rb.gy/xkmlh4](http://rb.gy/xkmlh4)
 """
 
 
-
-
-
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>⚠️Running on CPU, This may not work on CPU.</p>"
 
@@ -111,38 +101,41 @@ if torch.cuda.is_available():
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,
-        variant="fp16"
+        variant="fp16",
     )
     pipe2 = DiffusionPipeline.from_pretrained(
         "SG161222/RealVisXL_V2.02_Turbo",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,
-        variant="fp16"
+        variant="fp16",
     )
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()
         pipe2.enable_model_cpu_offload()
     else:
-        pipe.to(device)    
-        pipe2.to(device)    
+        pipe.to(device)
+        pipe2.to(device)
         print("Loaded on Device!")
-    
+
     if USE_TORCH_COMPILE:
         pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
         print("Model Compiled!")
 
+
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"
     img.save(unique_name)
     return unique_name
 
+
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     return seed
 
+
 @spaces.GPU(enable_queue=True)
 def generate(
     prompt: str,
@@ -159,14 +152,14 @@ def generate(
 ):
     if check_text(prompt, negative_prompt):
         raise ValueError("Prompt contains restricted words.")
-    
+
     prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
 
     if not use_negative_prompt:
         negative_prompt = ""  # type: ignore
-    negative_prompt += default_negative    
+    negative_prompt += default_negative
 
     options = {
         "prompt": prompt,
@@ -180,23 +173,24 @@ def generate(
         "use_resolution_binning": use_resolution_binning,
         "output_type": "pil",
     }
-    
+
     images = pipe(**options).images + pipe2(**options).images
 
     image_paths = [save_image(img) for img in images]
     return image_paths, seed
 
+
 examples = [
     "A closeup of a cat, a window, in a rustic cabin, close up, with a shallow depth of field, with a vintage film grain, in the style of Annie Leibovitz and in the style of Wes Anderson. --ar 85:128 --v 6.0 --style raw",
     "Daria Morgendorffer the main character of the animated series Daria, serious expression, very excites sultry look, so hot girl, beautiful charismatic girl, so hot shot, a woman wearing eye glasses, gorgeous figure, interesting shapes, life-size figures",
     "Dark green large leaves of anthurium, close up, photography, aerial view, in the style of unsplash, hasselblad h6d400c  --ar 85:128 --v 6.0 --style raw",
-    "Closeup of blonde woman depth of field, bokeh, shallow focus, minimalism, fujifilm xh2s with Canon EF lens, cinematic --ar 85:128 --v 6.0 --style raw"
+    "Closeup of blonde woman depth of field, bokeh, shallow focus, minimalism, fujifilm xh2s with Canon EF lens, cinematic --ar 85:128 --v 6.0 --style raw",
 ]
 
-css = '''
+css = """
 .gradio-container{max-width: 700px !important}
 h1{text-align:center}
-'''
+"""
 with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     gr.Markdown(DESCRIPTION)
     gr.DuplicateButton(
@@ -216,7 +210,9 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             run_button = gr.Button("Run")
         result = gr.Gallery(label="Result", columns=1, preview=True)
     with gr.Accordion("Advanced options", open=False):
-        use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
+        use_negative_prompt = gr.Checkbox(
+            label="Use negative prompt", value=True, visible=True
+        )
         negative_prompt = gr.Text(
             label="Negative prompt",
             max_lines=1,
@@ -241,12 +237,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 value=2,
             )
         seed = gr.Slider(
-            label="Seed",
-            minimum=0,
-            maximum=MAX_SEED,
-            step=1,
-            value=0,
-            visible=True
+            label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, visible=True
         )
         randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
         with gr.Row(visible=True):