Update app.py

app.py

@@ -211,40 +211,6 @@ def name2seq(file_name):
     images = torch.Tensor(images).unsqueeze(dim=0)
     return images
     
-    
-def display_gif(file_name, save_name):
-    images = []
-
-    for frame in range(8):
-        frame_name = '%d' % (frame)
-        image_filename = file_name + frame_name + '.png'
-        images.append(imageio.imread(image_filename))
-
-    gif_filename = 'avatar_source.gif'
-    return imageio.mimsave(gif_filename, images)
-
-
-def display_gif_pad(file_name, save_name):
-    images = []
-
-    for frame in range(8):
-        frame_name = '%d' % (frame)
-        image_filename = file_name + frame_name + '.png'
-        image = imageio.imread(image_filename)
-        image = image[:, :, :3]
-        image_pad = cv2.copyMakeBorder(image, 0, 0, 125, 125, cv2.BORDER_CONSTANT, value=0)
-        images.append(image_pad)
-
-    return imageio.mimsave(save_name, images)
-    
-
-def display_image(file_name):
-
-    image_filename = file_name + '0' + '.png'
-    print(image_filename)
-    image = imageio.imread(image_filename)
-    imageio.imwrite('image.png', image)
-    
 
 def concat(file_name):
     images = []
@@ -304,7 +270,7 @@ model.load_state_dict(torch.load('TransferVAE.pth.tar', map_location=torch.devic
 model.eval()
     
   
-def run(domain_source, 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, domain_target, action_target, hair_target, top_target, bottom_target):
 
     # == Source Avatar ==
     # body
@@ -422,7 +388,11 @@ def run(domain_source, action_source, hair_source, top_source, bottom_source, do
 gr.Interface(
     fn=run,
     inputs=[
-        gr.Textbox(value="Source Avatar - Human 😶", show_label=False, interactive=False),
+        gr.Markdown(
+            """
+            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"),