gokaygokay commited on
Commit
9576f72
1 Parent(s): d3e18f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,6 +9,7 @@ import subprocess
9
  import torch
10
  from PIL import Image
11
  from transformers import AutoProcessor, AutoModelForCausalLM
 
12
 
13
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
14
 
@@ -392,7 +393,7 @@ def create_interface():
392
  with gr.Row():
393
  with gr.Column(scale=2):
394
  with gr.Accordion("Basic Settings"):
395
- seed = gr.Number(label="Seed", value=0)
396
  custom = gr.Textbox(label="Custom Input Prompt (optional)")
397
  subject = gr.Textbox(label="Subject (optional)")
398
 
 
9
  import torch
10
  from PIL import Image
11
  from transformers import AutoProcessor, AutoModelForCausalLM
12
+ import random
13
 
14
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
15
 
 
393
  with gr.Row():
394
  with gr.Column(scale=2):
395
  with gr.Accordion("Basic Settings"):
396
+ seed = gr.Number(label="Seed", value=random.randint(0, 1000000), randomize=True)
397
  custom = gr.Textbox(label="Custom Input Prompt (optional)")
398
  subject = gr.Textbox(label="Subject (optional)")
399