Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
@@ -24,7 +25,7 @@ pipe.to("cuda")
 app = FaceAnalysis(name='buffalo_l')
 app.prepare(ctx_id=0, det_size=(640, 640))
 
-
+@spaces.GPU(enable_queue=True)
 def face_swap(src_img, dest_img):
     src_img = Image.open('./images/' + src_img + '.JPG')
     
@@ -53,7 +54,7 @@ def face_swap(src_img, dest_img):
 
     return final_image 
 
-
+@spaces.GPU(enable_queue=True)
 def greet(description,color,features,occasion,type_,face):
 
     # Parse input