First model version

app.py

@@ -49,16 +49,37 @@ def infer(filepath):
     visual_image = det_demo.visualization(image.copy(), result_polygons, result_masks, result_boxes, result_words)
     
     cv2.imwrite('result.jpg', visual_image)
-    return 'result.jpg', pd.DataFrame(result_words)
+    return ['result.jpg', pd.DataFrame(result_words)]
 
-description = gr.Markdown('''## Step 1 (Preprocess Input Image)
-- Drop an image containing a near-frontal face to the **Input Image**.
+blocks = gr.Blocks()
+
+input_image = gr.Image(label="image", type="filepath")
+output_image = gr.Image(label="out_img", type="filepath")
+output_word = gr.Dataframe(label="out_word", headers=['word'])
+
+with blocks:
+    description = gr.Markdown('''## Step 1 (Preprocess Input Image)
+    - Drop an image containing a near-frontal face to the **Input Image**.
     - If there are multiple faces in the image, hit the Edit button in the upper right corner and crop the input image beforehand.
-- Hit the **Detect & Align Face** button.
-- Hit the **Reconstruct Face** button.
+    - Hit the **Detect & Align Face** button.
+    - Hit the **Reconstruct Face** button.
     - The final result will be based on this **Reconstructed Face**. So, if the reconstructed image is not satisfactory, you may want to change the input image.
-''')
-
+    ''')
+    with gr.Row():
+        with gr.Column():
+            input_image.render()
+            button = gr.Button("Submit")
+            button.click(fn=infer, inputs=[input_image],
+                         outputs=[output_image, output_word],)
+        with gr.Column():
+            output_image.render()
+    with gr.Row():
+        output_word.render()
+    
+    
+if __name__ == "__main__":
+    blocks.launch(debug=True)
+'''
 iface = gr.Interface(
     fn=infer,
     title="张博强毕设展示",
@@ -66,4 +87,5 @@ iface = gr.Interface(
     inputs=[gr.inputs.Image(label="image", type="filepath")],
     outputs=[gr.outputs.Image(), gr.outputs.Dataframe(headers=['word'])],
     examples=['figs/test/CANDY.png', 'figs/test/ESPLANADE.png', 'figs/test/KAPPA.png'],
-).launch(enable_queue=True, cache_examples=True)
+).launch(enable_queue=True, cache_examples=True)
+'''