v1.6

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from PIL import Image
 from io import BytesIO
 from copy import deepcopy
-from src.core import process_inpaint
+from core import process_inpaint
 from huggingface_hub import login
 import os
 import spaces
@@ -15,14 +15,15 @@ def process_image(image):
     img_output = Image.fromarray(output).convert("RGB")
     return img_output
 
-demo = gr.Interface(
-    fn=process_image,
-    inputs=[
-        gr.ImageMask(type="numpy", layers=False, label="Upload Image"),
-    ],
-    outputs=gr.Image(format="png", label="Output Image"),
-    title="Object Remover",
-)
 
-if __name__ == "__main__":
-    demo.launch()
+with gr.Blocks() as demo:
+    gr.Markdown("# OBJECT REMOVER")
+    with gr.Row():
+        with gr.Column():
+            image = gr.ImageMask(type="numpy", layers=False, label="Upload Image")
+            button = gr.Button("Remove")
+        with gr.Column():
+            output = gr.Image(format="png", label="Output Image")
+        button.click(process_image, inputs=[image], outputs=[output])
+
+demo.launch()

src/core.py → core.py

@@ -20,7 +20,7 @@ from typing import Union
 
 import torch
 
-from src.helper import (
+from helper import (
     download_model,
     load_img,
     norm_img,