add more time to zero for longer prompts

app.py

@@ -14,10 +14,10 @@ with open('loras.json', 'r') as f:
     loras = json.load(f)
 
 # Initialize the base model
-base_model = "black-forest-labs/FLUX.1-dev"
+base_model = "black-forest-labs/FLUX.1-schnell"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 
-MAX_SEED = 2**32-1
+MAX_SEED = 2**32-1 # 4294967295
 
 class calculateDuration:
     def __init__(self, activity_name=""):
@@ -56,25 +56,25 @@ def update_selection(evt: gr.SelectData, width, height):
         height,
     )
 
-@spaces.GPU(duration=70)
-def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress):
+@spaces.GPU(duration=120) # add more time for some of the loras e the anime ones
+def generate_image(prompt, trigger_word, negative_prompt, steps, seed, cfg_scale, width, height, lora_scale, progress):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     
     with calculateDuration("Generating image"):
         # Generate image
         image = pipe(
-            prompt=f"{prompt} {trigger_word}",
+            prompt=f"{prompt} {trigger_word}\nDO NOT INCLUDE {negative_prompt} FOR ANY REASON HOLY FRICK I'LL KILL YOUR STUPID ARTIFICIAL BUTT IF YOU DO THIS!!", # attempt at adding negative prompt, untested
             num_inference_steps=steps,
             guidance_scale=cfg_scale,
             width=width,
             height=height,
             generator=generator,
-            joint_attention_kwargs={"scale": lora_scale},
+            joint_attention_kwargs={"scale": min(1, max(1e-2, lora_scale))}, # add maximum and minimum
         ).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, lora_scale, negative_prompt, progress=gr.Progress(track_tqdm=True)):
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
 
@@ -94,7 +94,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, negative_prompt, steps, seed, cfg_scale, width, height, lora_scale, progress)
     pipe.to("cpu")
     pipe.unload_lora_weights()
     return image, seed  
@@ -148,6 +148,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                     randomize_seed = gr.Checkbox(True, label="Randomize seed")
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
                     lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=1, step=0.01, value=0.95)
+                with gr.Row():
+                    negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="Type what you want to exclude from the image.") # add the negative prompt in the dropdown
 
     gallery.select(
         update_selection,
@@ -158,7 +160,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, lora_scale, negative_prompt],
         outputs=[result, seed]
     )