Update app.py

app.py

@@ -194,8 +194,8 @@ Demo is based of recent research from my Ph.D work. Results expects to be publis
 """
 
 iface = gradio.Interface(run_inference,
-                         [gradio.inputs.Image(shape=None, label='Target'),
-                          gradio.inputs.Image(shape=None, label='Source'),
+                         [gradio.inputs.Image(shape=None, type="pil", label='Target'),
+                          gradio.inputs.Image(shape=None, type="pil", label='Source'),
                           gradio.inputs.Slider(0, 100, default=100, label="Anonymization ratio (%)"),
                           gradio.inputs.Slider(0, 100, default=100, label="Adversarial defense ratio (%)"),
                           gradio.inputs.CheckboxGroup(["compare",
@@ -203,7 +203,7 @@ iface = gradio.Interface(run_inference,
                                                        "reconstruction attack",
                                                        "adversarial defense"],
                                                       label='Options')],
-                         gradio.outputs.Image(),
+                         gradio.outputs.Image(type="pil"),
                          title="Face Swap",
                          description=description,
                          examples=examples,