Update app.py

app.py

@@ -2,6 +2,7 @@ import subprocess
 import os
 import gradio as gr
 import torch
+import spaces
 
 if torch.cuda.is_available():
     device = "cuda"
@@ -14,6 +15,7 @@ else:
 subprocess.run(["git", "clone", "https://github.com/Nick088Official/Stable_Diffusion_Finetuned_Minecraft_Skin_Generator.git"])
 os.chdir("Stable_Diffusion_Finetuned_Minecraft_Skin_Generator")
 
+@spaces.GPU()
 def run_inference(prompt, stable_diffusion_model, num_inference_steps, guidance_scale, model_precision_type, seed, output_image_name, verbose):
     if stable_diffusion_model == '2':
         sd_model = "minecraft-skins"