Cast paths to str

app.py

@@ -179,7 +179,7 @@ class ImageEditor(object):
                 out_img = torch.cat(img_list, axis=0)
                 utils.save_image(out_img, out_path, nrow=int(np.sqrt(out_img.size(0))), normalize=True, scale_each=True, range=(-1, 1))
 
-            return Path(out_path)
+            return str(out_path)
         
         return self.generate_vid(generators, inverted_latent, out_dir, video_format, with_editing)
 
@@ -206,7 +206,7 @@ class ImageEditor(object):
             else:
                 shutil.copy2(gen_path, out_path) 
 
-        return out_path
+        return str(out_path)
 
     def run_alignment(self, image_path):
         aligned_image = align_face(filepath=image_path, predictor=self.shape_predictor)
@@ -284,17 +284,17 @@ with blocks:
             with gr.Column():
                 img_button = gr.Button("Edit Image")
             with gr.Column():
-                img_output = gr.outputs.Image(type="file")
+                img_output = gr.outputs.Image(type="file", label="Output Image")
             
         with gr.TabItem("Create Video"):
             with gr.Column():
                 with gr.Row():
+                    vid_button = gr.Button("Generate Video")
                     edit_choice       = gr.inputs.Checkbox(default=False, label="With latent space editing?", optional=False)
                     vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format")
-                    vid_button = gr.Button("Generate Video")
             
             with gr.Column():
-                vid_output = gr.outputs.Video()
+                vid_output = gr.outputs.Video(label="Output Video")
 
     img_button.click(fn=editor.edit_image, inputs=[input_img, style_choice], outputs=img_output)
     vid_button.click(fn=editor.edit_video, inputs=[input_img, style_choice, edit_choice, vid_format_choice], outputs=vid_output)