Update app.py

app.py

@@ -43,8 +43,9 @@ title = "Undress AI"
 description = "⛔ Input photos of people, similar to the test picture at the bottom, and undress pictures will be produced. You may have to wait 30 seconds for a picture. 🔞 Do not upload personal photos 🔞 There is a queue system. According to the logic of first come, first served, only one picture will be made at a time. Must be able to at least see the outline of a human body ⛔"
 
 examples = [
-    ['input.png', 'Test'],
-    ['input.jpg', 'Test'],
+    ['example1.png'],
+    ['example2.png'],
+    ['example3.png'],
 ]
 
 css = """
@@ -58,24 +59,21 @@ body {
 }
 footer {display: none !important;}
 """
-
+    
 with gr.Blocks(css=css) as demo:
-    width=240
-    height=320
+    url_params = gr.State()
+    demo.load(get_params, None, url_params)
+    url_params = url_params.value
+    width=320
+    height=240
     with gr.Row():
         with gr.Column(scale=2):  # Adjust scale for proper sizing
-            image_input = gr.Image(type="numpy", label="Upload Image", width=width, height=height,scale=1)
-            process_button = gr.Button("Nude!",scale=1)
-            status = gr.Markdown(value="")
-        with gr.Column(scale=1):  # Adjust scale for proper sizing
-            gr.Markdown("### Examples")
-            example_1 = gr.Examples(examples=[examples[0]], inputs=image_input, examples_per_page=1)
-            example_2 = gr.Examples(examples=[examples[1]], inputs=image_input, examples_per_page=1)
+            image_input = gr.Image(type="numpy", label="Upload Image", width=width, height=height)
+            gr.Examples(examples=examples, inputs=image_input, examples_per_page=3)
+            process_button = gr.Button("Nude!").style(width=width)
 
     def update_status(img):
-        status.update(value="Processing... Please wait.")
         processed_img = inference(img)
-        status.update(value="Processing complete!")
         return processed_img
 
     process_button.click(update_status, inputs=image_input, outputs=image_input)