Update app.py
Browse files
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 |
-
|
| 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 |
-
|
| 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"),
|