Update app.py

app.py

@@ -34,7 +34,7 @@ def process(image, mask):
     mask_bytes = mask.convert("RGB").tobytes()
     
     # Gọi API với hình ảnh và mask (token được cố định trong code)
-    access_token = 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJjbGllbnRfaWQiOiIyRGZQSm1sY1BLalBuaGh4ajBxSVpmQktfbS12RFhTc1NMRFNOX1gtZlVVIiwiZXhwIjoxNzI3MTY0MTcyLCJhcHBfbmFtZSI6IjE3NTMwODQiLCJtZXRhZGF0YSI6bnVsbCwiZ3JhbnRfdHlwZSI6ImNyZWQifQ.24oHGH9ial_gqeYrpUbYP68MwYly2ArzXbhbE1hGzQo'
+    access_token = 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9...'
     
     result = erase_object(image_bytes, mask_bytes, access_token)
     
@@ -46,9 +46,9 @@ with gr.Blocks() as demo:
 
     # Input cho ảnh và Sketchpad để vẽ mask
     with gr.Row():
-        image_input = gr.Image(type="pil", label="Tải ảnh lên")
-        mask_input = gr.Sketchpad(shape=(512, 512), label="Vẽ mask để xóa đối tượng")
-
+        # Cho phép tải lên hình ảnh và vẽ trực tiếp mask lên ảnh đó
+        image_input = gr.Image(type="pil", tool="sketch", label="Tải ảnh lên và vẽ mask")
+    
     # Nút thực hiện xử lý xóa đối tượng
     submit_button = gr.Button("Xóa đối tượng")
 
@@ -58,7 +58,7 @@ with gr.Blocks() as demo:
     # Kết nối các thành phần với nhau
     submit_button.click(
         process,
-        inputs=[image_input, mask_input],
+        inputs=[image_input, image_input],  # Sử dụng cùng một ảnh cho cả đầu vào và mask
         outputs=result_output
     )