angelahzyuan commited on
Commit
7d08c43
·
verified ·
1 Parent(s): 8315dde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -9
app.py CHANGED
@@ -7,14 +7,6 @@ import spaces
7
 
8
 
9
 
10
- def set_seed(seed=5775709):
11
- random.seed(seed)
12
- np.random.seed(seed)
13
- torch.manual_seed(seed)
14
- torch.cuda.manual_seed(seed)
15
-
16
-
17
- set_seed()
18
 
19
  if torch.cuda.is_available():
20
  pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16")
@@ -47,7 +39,7 @@ with gr.Blocks() as demo:
47
  with gr.Row():
48
  prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Type something...", lines=2)
49
  generate_btn = gr.Button("Generate images")
50
- guidance_scale = gr.Slider(label="Guidance Scale", minimum=0, maximum=50, value=9, step=0.1)
51
  num_images_input = gr.Number(label="Number of images", value=5, minimum=1, maximum=10, step=1)
52
  gallery = gr.Gallery(label="Generated images", elem_id="gallery", columns=5, object_fit="contain")
53
 
 
7
 
8
 
9
 
 
 
 
 
 
 
 
 
10
 
11
  if torch.cuda.is_available():
12
  pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16")
 
39
  with gr.Row():
40
  prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Type something...", lines=2)
41
  generate_btn = gr.Button("Generate images")
42
+ guidance_scale = gr.Slider(label="Guidance Scale", minimum=0, maximum=50, value=7.5, step=0.1)
43
  num_images_input = gr.Number(label="Number of images", value=5, minimum=1, maximum=10, step=1)
44
  gallery = gr.Gallery(label="Generated images", elem_id="gallery", columns=5, object_fit="contain")
45