recompile

app.py

@@ -1,15 +1,21 @@
 from subprocess import Popen, PIPE, STDOUT, check_output
 import tempfile
 import gradio as gr
+from PIL import Image
 
 def run(input_image):
     output = check_output(["chmod", "a+x", "bin/detect-image"])
-    cmd = 'bin/detect-image ' + input_image
-    p = Popen(cmd, shell=True, stdin=PIPE, stdout=PIPE, stderr=STDOUT, close_fds=True)
-    return p.stdout.read()
+
+    with tempfile.TemporaryDirectory() as tmpdir:
+        output_image_filename = tmpdir + "/result.jpg"
+        cmd = 'bin/detect-image ' + input_image + ' ' + output_image_filename
+        p = Popen(cmd, shell=True, stdin=PIPE, stdout=PIPE, stderr=STDOUT, close_fds=True)
+        i = Image.open(output_image_filename)
+
+    return p.stdout.read(), i
 
 gr.Interface(
     fn=run,
     inputs=gr.Image(type="filepath", label="Input Image"),
-    outputs=gr.Textbox(),
+    outputs=[gr.Image(type="pil"), gr.Textbox()],
 ).launch()