Update app.py

app.py

@@ -57,5 +57,20 @@ def main(image):
 
 
 # Gradio
-iface = gr.Interface(fn=main, inputs=[gr.Image(type='pil', image_mode='L')], outputs='text')
-iface.launch()
+with gr.Blocks(title="Aging Biomarker from CXR",
+            css=".gradio-container {background:mintcream;}"
+        ) as demo:
+    gr.HTML("""<div style="font-family:'Times New Roman', 'Serif'; font-weight:bold; text-align:center; color:royalblue;">Aging Biomarker from CXR</div>""")
+
+    with gr.Row(): 
+        input_image = gr.Image(type="pil", image_mode="L", shape=(320, 320))        
+        output_label=gr.Label(label="Estimated age")
+
+    send_btn = gr.Button("Inference")
+    send_btn.click(fn=main, inputs=input_image, outputs=output_label)
+
+    with gr.Row():
+        gr.Examples(['./samples/66_female_xp.png'], label='CXR: 66 years old female', inputs=input_image)
+        gr.Examples(['./samples/28_male_xp.png'], label='CXR: 28 years old male', inputs=input_image)
+
+demo.launch(debug=True)