Update app.py

app.py

@@ -60,8 +60,8 @@ description = "This demo introduces an interactive playground for our trained De
 
 gr.Interface(
     inference, 
-    [gr.inputs.Image(type="pil", label="Input")], 
-    gr.outputs.Image(type="numpy", label="Output"),
+    gr.Image(type="pil", label="Input"), 
+    gr.Label(label="Prediction"),
     title=title,
     description=description,
     examples=[]).launch()