Update app.py

app.py

@@ -27,14 +27,22 @@ class Generator(nn.Module):
         return output
 
 
-def predict(body, hair, top, bottom):
-    name = str(body) + str(hair) + str(top) + str(bottom)
+def path(action, body, hair, top, bottom):
+
+    # body
+    if body == "human": body = '0'
+    elif body == "alien": body = '1'
+    
+    # hair
+
+    name = action + str(body) + str(hair) + str(top) + str(bottom)
     return name
 
 
 gr.Interface(
-    predict,
+    path,
     inputs=[
+        gr.Radio(["shoot", "slash", "spellcard", "thrust", "walk"]),
         gr.Radio(["human", "alien"]),
         gr.Slider(0, 5, label='Hair', step=1, default=0),
         gr.Slider(0, 3, label='Top', step=1, default=0),