Update app.py

app.py

@@ -4,6 +4,7 @@
 # @Author  : ashui(Binghui Chen)
 from sympy import im
 import time
+import spaces
 import cv2
 import gradio as gr
 import numpy as np
@@ -52,7 +53,8 @@ showcases = []
 for idx, name in enumerate(files):
         temp = os.path.join(os.path.dirname(__file__), img, name)
         showcases.append(temp)
-
+    
+@spaces.GPU(enable_queue=True)
 def process(original_image, original_mask, input_mask, selected_points, prompt,negative_prompt,guidance_scale,steps,strength,scheduler):
     if original_image.shape[0]>original_image.shape[1]:
         original_image=cv2.resize(original_image,(int(original_image.shape[1]*1000/original_image.shape[0]),1000))
@@ -126,7 +128,7 @@ block = gr.Blocks(
              radius_size=gr.themes.sizes.radius_none,
              text_size=gr.themes.sizes.text_md
          )
-        ).queue(concurrency_count=2)
+        ).queue(concurrency_count=2,max_size=4)
 with block:
     with gr.Row():
         with gr.Column():
@@ -307,5 +309,5 @@ with block:
     ips=[original_image, original_mask, input_mask, selected_points, prompt,negative_prompt,guidance_scale,steps,strength,scheduler]
     run_button.click(fn=process, inputs=ips, outputs=[result_gallery, request_id, gallery_flag])
 
-
 block.launch()
+