Update app.py

app.py

@@ -6,6 +6,7 @@ import cv2
 import numpy as np
 import torch
 from huggingface_hub import hf_hub_download
+import spaces
 
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 print(f"Is CUDA available: {torch.cuda.is_available()}")
@@ -65,10 +66,14 @@ def transform_image_bbox(prompts):
 
     return img, bbox, (ori_H, ori_W), (new_H, new_W)
 
+
+
+
 if __name__ == '__main__':
 
     model, model_type = load_model()
 
+    @spaces.GPU
     def inference_image(prompts, input_model_type):
 
         global model_type