Update MeAI_Maincode/Polyb.py

MeAI_Maincode/Polyb.py

@@ -91,12 +91,11 @@ def main(image):
     return highlighted_image  # Return the highlighted image
     
 def create_polyb():
-    demo = gr.Interface(
-        fn=main,
-        inputs=gr.Image(label = "Xin Nhập Ảnh Vào", type = 'pil', height=512, width=512,
-          value="/anh/braintt.jpg",interactive=True),
-        outputs=gr.Image(label = "Kết Quả", type = 'pil', height=512, width=512),
-        title="Highlight Polyps",
-        description="Tải lên một hình ảnh và phát hiện polyp với LangSAM."
-    )  
-    return demo
+  with gr.Blocks() as demo:
+      gr.Markdown("Hãy tải ảnh lên và nhấn **Xử Lý** để khoanh vùng Polyp.")
+      with gr.Row():
+          inp = gr.Image(label= "Nhập Ảnh",type="pil",height=512, width=512,value="../anh/viemphoi.jpeg"),interactive=True)
+          out = gr.Label(label="Kết Quả Dự Đoán", type = 'pil', height=512, width=512)
+      btn = gr.Button("Xử Lý")
+      btn.click(fn=main, inputs=inp, outputs=out)
+  return demo