Update app.py

app.py

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