Update app.py

app.py

@@ -104,7 +104,7 @@ def update_selection(evt: gr.SelectData, width, height):
     )
 
 @spaces.GPU(duration=70)
-def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress):
+def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
@@ -117,11 +117,12 @@ def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height,
             width=width,
             height=height,
             generator=generator,
+            negative_prompt=negative_prompt,
             joint_attention_kwargs={"scale": lora_scale},
         ).images[0]
     return image
 
-def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, negative_prompt, lora_scale, progress=gr.Progress(track_tqdm=True)):
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
 
@@ -151,7 +152,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
     
-    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress)
+    image = generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, negative_prompt, lora_scale, progress)
     pipe.to("cpu")
     pipe.unload_lora_weights()
     return image, seed  
@@ -167,22 +168,24 @@ css = '''
 '''
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     title = gr.HTML(
-        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> SOONfactory </h1>""",
+        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> LibreFLUX SOONfactory </h1>""",
         elem_id="title",
     )
     	    # Info blob stating what the app is running
     info_blob = gr.HTML(
-        """<div id="info_blob"> Novorealist LoRa-stocked Birthweek-inspired Img Manufactory for Dunova, Dunovas, & Dunovaists!</div>"""
+        """<div id="info_blob"> SOON®'s curated LoRa Gallery & Art Manufactory Space.|Runs on LibreFLUX model +  Zer0int's fine-tuned CLIP (*'normal' 77 tokens)| Largely stocked w/our trained LoRAs: Historic Color, Silver Age Poets, Sots Art, more!|</div>"""
     )
 
         # Info blob stating what the app is running
     info_blob = gr.HTML(
-        """<div id="info_blob">Trigger LoRAs by Pre-phrasing Prompts w/: 1-5. ADU person (/'ADU woman') photo |6-15. HST style |16. how2draw |17-20.HST |21. HST Austin Osman Spare style |22. RCA |23. propaganda poster |24. SOTS art |25. pficonics |26. wh3r3sw4ld0 |27. vintage cover |28. crisp photo |29. retrofuturism |30. Film Photo |31. TOK hybrid |32. 2004 photo |33. Unexpected photo |34. flmft |35. TOK portra |36. Yearbook photo |37. Akhmatova |38. Tsvetaeva |39. Blok |40. LEN Lenin |41. Trotsky |42. Rosa Luxemburg </div>"""
+        """<div id="info_blob"> Pre-phrase Prompts w/: 1-2. HST style |3. RCA poster |4.SOTS art |5.HST Austin Osman Spare |6. Mayakovsky |7-8. Tsvetaeva |9. Akhmatova |10. Mandelshtam |11-13. Blok |14. LEN Lenin |15. Trotsky |16. Rosa Fluxenburg |17-30. HST |31. how2draw |32. propaganda poster |33. TOK photo cartoon hybrid |34. photo |35.unexpected photo |36. flmft |37. Yearbook |38. TOK portra |39. pficonics |40. retrofuturism |41. wh3r3sw4ld0 |42. amateur photo |43. crisp photo |44-45. ADU |46. Film Photo |47. ff-collage |48. HST|49-50. AOS Austin Osman Spare art |51. cover </div>"""
     )
     selected_index = gr.State(None)
     with gr.Row():
-        with gr.Column(scale=3):
+        with gr.Column(scale=2):
             prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Select LoRa/Style & type prompt!")
+        with gr.Column(scale=2):
+            negative_prompt = gr.Textbox(label="Negative Prompt", lines=1, placeholder="What to exclude!")
         with gr.Column(scale=1, elem_id="gen_column"):
             generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
     with gr.Row():
@@ -208,7 +211,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                 
                 with gr.Row():
                     width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
-                    height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1088)
+                    height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024)
                 
                 with gr.Row():
                     randomize_seed = gr.Checkbox(True, label="Randomize seed")
@@ -224,7 +227,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
-        inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
+        inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, negative_prompt, lora_scale],
         outputs=[result, seed]
     )