Update app.py

app.py

@@ -25,7 +25,6 @@ PATH = {
 
 prompt_types = ["Point", "Box", "Mask"]
 
-@spaces.GPU()
 def load_3d_scene(name, sample_idx=-1, type_=None, prompt=None, final=False, new_color=None):
    DATASET = name.split('-')[1].replace(" ", "")
    path = 'data/' + DATASET + '/' + PATH[DATASET][sample_idx]
@@ -198,13 +197,11 @@ def start_segmentation(name=None, sample_idx=None, prompt_type=None, prompt_idx=
    fig = load_3d_scene(name, sample_idx, TYPE, prompt, final=True, new_color=new_color)
    return fig, gr.Textbox(label="Response", value="Segmentation completed successfully!", visible=True)
 
-@spaces.GPU()
 def update1(datasets):
    if 'Objaverse' in datasets:
        return gr.Radio(label="Select 3D Object", choices=samples[datasets]), gr.Textbox(label="Response", value="", visible=True) 
    return gr.Radio(label="Select 3D Scene", choices=samples[datasets]), gr.Textbox(label="Response", value="", visible=True) 
 
-@spaces.GPU()
 def update2(name, sample_idx, prompt_type):
    if name == None or sample_idx == None or prompt_type == None:
        return gr.Radio(label="Select Prompt Example", choices=[]), gr.Textbox(label="Response", value="", visible=True) 
@@ -225,7 +222,6 @@ def update2(name, sample_idx, prompt_type):
    cur = ['Example ' + str(i) for i in range(1, len(info) + 1)]
    return gr.Radio(label="Select Prompt Example", choices=cur), gr.Textbox(label="Response", value="", visible=True) 
     
-@spaces.GPU()
 def update3(name, sample_idx, prompt_type, prompt_idx):
    if name == None or sample_idx == None or prompt_type == None:
        return gr.Textbox(label="Response", value="", visible=True), gr.Slider(minimum=0.01, maximum=0.15, step=0.001, label="Voxel Size", value=0.02)
@@ -243,7 +239,6 @@ def update3(name, sample_idx, prompt_type, prompt_idx):
   
    return gr.Textbox(label="Response", value="", visible=True), gr.Slider(minimum=0.01, maximum=0.15, step=0.001, label="Voxel Size", value=vx_)
 
-@spaces.GPU()
 def main():
    title = """<h1 style="text-align: center;">
     <div style="width: 1.2em; height: 1.2em; display: inline-block;"><img src="https://github.com/ZiyuGuo99/ZiyuGuo99.github.io/blob/main/assets/img/logo.png?raw=true" style='width: 100%; height: 100%; object-fit: contain;' /></div>
@@ -302,7 +297,8 @@ def main():
        start_segment_button.click(start_segmentation, inputs=[sample_dropdown, scene_dropdown, prompt_type_dropdown, prompt_sample_dropdown], outputs=[plot1, response])
   
    #app.queue(status_update_rate="auto")
-   app.launch()
+   app.queue(max_size=20, api_open=False)
+   app.launch(max_threads=400)
 
 if __name__ == "__main__":
    main()