ldkong commited on
Commit
ad44f18
β€’
1 Parent(s): a5b7c29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -270,7 +270,7 @@ model.load_state_dict(torch.load('TransferVAE.pth.tar', map_location=torch.devic
270
  model.eval()
271
 
272
 
273
- def run(action_source, hair_source, top_source, bottom_source, action_target, hair_target, top_target, bottom_target):
274
 
275
  # == Source Avatar ==
276
  # body
@@ -390,7 +390,7 @@ gr.Interface(
390
  inputs=[
391
  gr.Markdown(
392
  """
393
- Source Avatar - Human πŸ‘¦πŸ»
394
  """
395
  ),
396
  gr.Radio(choices=["slash", "spellcard", "walk"], value="slash"),
@@ -399,7 +399,7 @@ gr.Interface(
399
  gr.Radio(choices=["white", "golden", "red", "silver"], value="white"),
400
  gr.Markdown(
401
  """
402
- Target Avatar - Alien πŸ‘½
403
  """
404
  ),
405
  gr.Radio(choices=["slash", "spellcard", "walk"], value="walk"),
 
270
  model.eval()
271
 
272
 
273
+ def run(source, action_source, hair_source, top_source, bottom_source, target, action_target, hair_target, top_target, bottom_target):
274
 
275
  # == Source Avatar ==
276
  # body
 
390
  inputs=[
391
  gr.Markdown(
392
  """
393
+ πŸ‘¦πŸ» Human - Source Avatar
394
  """
395
  ),
396
  gr.Radio(choices=["slash", "spellcard", "walk"], value="slash"),
 
399
  gr.Radio(choices=["white", "golden", "red", "silver"], value="white"),
400
  gr.Markdown(
401
  """
402
+ πŸ‘½ Alien - Target Avatar
403
  """
404
  ),
405
  gr.Radio(choices=["slash", "spellcard", "walk"], value="walk"),