Update app.py

app.py

@@ -51,26 +51,27 @@ css = """
 body {
     background-color: rgb(17, 24, 39);
     color: white;
+    overflow: hidden; /* Prevent scrolling */
 }
 .gradio-container {
     background-color: rgb(17, 24, 39) !important;
     border: none !important;
+    max-width: 100%; /* Ensure it does not exceed the container's width */
+    max-height: 100%; /* Ensure it does not exceed the container's height */
+    overflow: hidden; /* Prevent internal scrolling */
 }
-footer {display: none !important;}
+footer {display: none !important;} /* Hide footer */
 """
 
 with gr.Blocks(css=css) as demo:
     with gr.Column():
-        image_input = gr.Image(type="numpy", label="Upload Image")
+        image_input = gr.Image(type="numpy", label="Upload Image", height=512, width=512)
         process_button = gr.Button("Process Image")
         status = gr.Markdown(value="")
 
     def update_status(img):
-        status.update(value="Processing... Please wait.")
-        processed_img = inference(img)
-        status.update(value="Processing complete!")
-        return processed_img
+        return inference(img), gr.update(value="Processing complete!")
 
-    process_button.click(update_status, inputs=image_input, outputs=image_input)
+    process_button.click(update_status, inputs=image_input, outputs=[image_input, status])
 
 demo.launch()