Update app.py

app.py

@@ -270,7 +270,7 @@ model.load_state_dict(torch.load('TransferVAE.pth.tar', map_location=torch.devic
 model.eval()
     
   
-def run(action_source, hair_source, top_source, bottom_source, domain_target, action_target, hair_target, top_target, bottom_target):
+def run(action_source, hair_source, top_source, bottom_source, action_target, hair_target, top_target, bottom_target):
 
     # == Source Avatar ==
     # body
@@ -390,14 +390,18 @@ gr.Interface(
     inputs=[
         gr.Markdown(
             """
-            Source Avatar - Human 😶
+            Source Avatar - Human 👦🏻
             """
         ),
         gr.Radio(choices=["slash", "spellcard", "walk"], value="slash"),
         gr.Radio(choices=["green", "yellow", "rose", "red", "wine"], value="green"),
         gr.Radio(choices=["brown", "blue", "white"], value="brown"),
         gr.Radio(choices=["white", "golden", "red", "silver"], value="white"),
-        gr.Textbox(value="Target Avatar - Alien 👽", show_label=False, interactive=False),
+        gr.Markdown(
+            """
+            Target Avatar - Alien 👽
+            """
+        ),
         gr.Radio(choices=["slash", "spellcard", "walk"], value="walk"),
         gr.Radio(choices=["violet", "silver", "purple", "grey", "golden"], value="golden"),
         gr.Radio(choices=["grey", "khaki", "linen", "ocre"], value="ocre"),