Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import sys
-sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+import spaces
 
 import gradio as gr
 import torch
@@ -425,6 +425,7 @@ def load_model_and_vae(model_path, vae_path):
         CURRENT_MODEL_PATH = None
         return f"✗ Failed: {str(e)}"
 
+@spaces.GPU()
 def generate_wrapper(
     input_image, prompt_text, model_path, vae_path, height, width,
     text_steps, text_gen_length, text_block_length, cfg_scale, cfg_img,