srijaydeshpande commited on
Commit
8b9cf54
·
verified ·
1 Parent(s): 5c61b5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +27,7 @@ DEFAULT_NEGATIVE_PROMPT = (
27
 
28
 
29
  @spaces.GPU
30
- def run(bucket_folder, person_name, req_id):
31
 
32
  aws_access_key_id = 'AKIA2NMAMYX4K55CZ7HR'
33
  BUCKET = 'syntheticai-headshots'
@@ -200,7 +200,7 @@ with gr.Blocks(title="AI headshot Generator", css=".gr-box {border-color: #8136e
200
 
201
  demo = gr.Interface(
202
  run,
203
- inputs=['textbox', 'textbox', 'textbox'],
204
  outputs=[output, intermediate_output],
205
  title="AI Headshot Diffusor"
206
  )
 
27
 
28
 
29
  @spaces.GPU
30
+ def run(bucket_folder, person_name, req_id, gender='male'):
31
 
32
  aws_access_key_id = 'AKIA2NMAMYX4K55CZ7HR'
33
  BUCKET = 'syntheticai-headshots'
 
200
 
201
  demo = gr.Interface(
202
  run,
203
+ inputs=['textbox', 'textbox', 'textbox', 'textbox'],
204
  outputs=[output, intermediate_output],
205
  title="AI Headshot Diffusor"
206
  )