nroggendorff commited on
Commit
540be8a
·
verified ·
1 Parent(s): c69b53e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  import spaces
 
3
 
4
  import torch
5
  from diffusers import FluxPipeline
@@ -14,7 +15,7 @@ def generate(prompt, negative_prompt, width, height, sample_steps, lora_id):
14
  except RepositoryNotFoundError:
15
  raise ValueError(f"Recieved invalid FLUX LoRA.")
16
 
17
- return pipeline(prompt=f"{prompt}\nDO NOT INCLUDE {negative_prompt}", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(42), guidance_scale=7).images[0]
18
 
19
  with gr.Blocks() as interface:
20
  with gr.Column():
 
1
  import gradio as gr
2
  import spaces
3
+ import random
4
 
5
  import torch
6
  from diffusers import FluxPipeline
 
15
  except RepositoryNotFoundError:
16
  raise ValueError(f"Recieved invalid FLUX LoRA.")
17
 
18
+ return pipeline(prompt=f"{prompt}\nDO NOT INCLUDE {negative_prompt}", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(random.randint(42, 69)), guidance_scale=7).images[0]
19
 
20
  with gr.Blocks() as interface:
21
  with gr.Column():