Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from huggingface_hub import login, HfFileSystem, HfApi, ModelCard
 import os 
+import random
+import spaces
 
 is_shared_ui = True if "fffiloni/sd-xl-custom-model" in os.environ['SPACE_ID'] else False
 hf_token = os.environ.get("HF_TOKEN")
@@ -53,7 +55,7 @@ def load_model(custom_model):
     return gr.update(choices=sfts_available_files, value=sfts_available_files[0], visible=True), gr.update(value=instance_prompt, visible=True)
 
 
-
+@spaces.GPU
 def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
 
     if weight_name == "NO SAFETENSORS FILE": 
@@ -206,7 +208,8 @@ with gr.Blocks(css=css) as demo:
     load_model_btn.click(
         fn = load_model,
         inputs=[custom_model],
-        outputs = [model_status, weight_name, trigger_word]
+        outputs = [model_status, weight_name, trigger_word],
+        queue = False
     )
     submit_btn.click(
         fn = infer,