Update app.py

app.py

@@ -251,8 +251,8 @@ def main_fun(image_pil, x_units, y_units, text_prompt, box_threshold, text_thres
     output = inpainted_shifted + shifted_image
     output = Image.fromarray(output)
 #     visualize_results(image_pil, output, 'shifted')
-    
-    return segmented_image, output
+    segmented_image = Image.fromarray(segmented_image)
+    return segmented_image.resize((512, 512)), output.resize((512, 512))
 
 import gradio as gr
 
@@ -263,9 +263,9 @@ with image_blocks as demo:
         with gr.Column():
             image = gr.Image(sources=['upload'], type="pil", label="Upload")
             # with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-            text_prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=True)
-            x_units = gr.Slider(minimum=50, maximum=300, step=10, value=10, label="x_units")
-            y_units = gr.Slider(minimum=30, maximum=300, step=10, value=5, label="y_units")
+            text_prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', label="Object class", show_label=True)
+            x_units = gr.Slider(minimum=0, maximum=300, step=10, value=100, label="x_units")
+            y_units = gr.Slider(minimum=0, maximum=300, step=10, value=0, label="y_units")
             with gr.Accordion("Advanced options", open=False) as advanced_options:
                 box_threshold = gr.Slider(
                     label="Box Threshold", minimum=0.0, maximum=1.0, value=0.23, step=0.01