Update app.py

app.py

@@ -103,7 +103,7 @@ class ImageEditor(object):
 
         model_path = self.experiment_args["model_path"]
 
-        ckpt = torch.load(model_path, map_location="cpu")
+        ckpt = torch.load(model_path, map_location="cuda:0" if torch.cuda.is_available() else "cpu")
         opts = ckpt["opts"]
 
         opts["checkpoint_path"] = model_path
@@ -229,11 +229,10 @@ with blocks:
         "<div>Upload an image of your face, pick your desired output styles, pick any modifiers, and apply StyleGAN-based editing.</div>"
     )
     with gr.Row():
+	    with gr.Column():
+            input_img = gr.Image(type="filepath", label="Input image")
         with gr.Column():
-            input_img = gr.inputs.Image(type="filepath", label="Input image")
-
-        with gr.Column():
-            style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), default=editor.get_style_list(), type="value",
+            style_choice = gr.CheckboxGroup(choices=editor.get_style_list(), default=editor.get_style_list(), type="value",
                                                    label="Styles")
             alter = gr.Dropdown(
                 choices=["None", "Masculine", "Feminine", "Smiling", "Frowning", "Young", "Old", "Short Hair",