Update app.py

app.py

@@ -1,3 +1,5 @@
+import spaces
+
 import time
 
 from diffusers import AutoPipelineForInpainting
@@ -12,6 +14,7 @@ import gradio as gr
 from gradio import components
 import difflib
 
+
 # Constants
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 
@@ -34,6 +37,7 @@ def get_most_similar_string(target_string, string_array):
     
     return best_match
 
+
 def loadModels():
 
     yoloModel=YOLO('yolov8x-seg.pt')
@@ -99,6 +103,7 @@ def getSegments(yoloModel,img1):
 
 # Gradio UI
 
+@spaces.GPU
 def getDescript(image_captioner,img1):
     base64_img = image_to_base64(img1)
     caption = image_captioner(base64_img)[0]['generated_text']
@@ -125,6 +130,7 @@ def ChangeOBJ(sdxl_m,img1,response,mask1):
 
 yoloModel,sdxl,image_captioner=loadModels()
 
+@spaces.GPU
 def full_pipeline(image, target):
     img1 = Image.fromarray(image.astype('uint8'), 'RGB')
     allMask=getSegments(yoloModel,img1)