DarianT commited on
Commit
9053798
·
1 Parent(s): ec285c2

Add previous id functionality

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -31,9 +31,10 @@ gender_dict = {"ID_1": "male", "ID_5": "male", "ID_16": "female", "ID_20": "male
31
  MAX_SEED = 10000
32
  image_size = 512
33
 
34
-
35
  previous_id = ""
36
 
 
37
  @spaces.GPU # Uncomment if using ZeroGPU
38
  def infer(
39
  identity,
@@ -44,7 +45,6 @@ def infer(
44
  randomize_seed,
45
  guidance_scale,
46
  num_inference_steps,
47
- progress=gr.Progress(track_tqdm=True),
48
  num_images=1
49
  ):
50
 
@@ -87,6 +87,7 @@ css = """
87
  """
88
 
89
 
 
90
  with gr.Blocks(css=css) as demo:
91
  with gr.Column(elem_id="col-container"):
92
  gr.Markdown(" # ID-Booth Demo")
 
31
  MAX_SEED = 10000
32
  image_size = 512
33
 
34
+ global previous_id
35
  previous_id = ""
36
 
37
+
38
  @spaces.GPU # Uncomment if using ZeroGPU
39
  def infer(
40
  identity,
 
45
  randomize_seed,
46
  guidance_scale,
47
  num_inference_steps,
 
48
  num_images=1
49
  ):
50
 
 
87
  """
88
 
89
 
90
+
91
  with gr.Blocks(css=css) as demo:
92
  with gr.Column(elem_id="col-container"):
93
  gr.Markdown(" # ID-Booth Demo")