Update app.py

app.py

@@ -20,8 +20,13 @@ def ocr(image):
 
     return output
 
+def process_output(output):
+    output["step1_image"] = Image.fromarray(output["step1_image"])
+    return output
+
 iface = gr.Interface(fn=ocr,
-                     inputs=gr.inputs.Image(type="pil"),
+                     inputs=gr.Image(type="pil"),
                      outputs=["text", "image", "text", "text"])
 
+iface.process_output = process_output
 iface.launch()