update gpu

app.py

@@ -145,13 +145,13 @@ class MobileSAM(nn.Module):
 
 mobilesam = MobileSAM()
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=60)
 def image_mobilesam_feature(
     images,
     node_type="block",
     layer=-1,
 ):
-
+    global USE_CUDA
     if USE_CUDA:
         images = images.cuda()
 
@@ -249,13 +249,13 @@ class SAM(torch.nn.Module):
 
 sam = SAM()
 
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=120)
 def image_sam_feature(
     images,
     node_type="block",
     layer=-1,
 ):
-
+    global USE_CUDA
     if USE_CUDA:
         images = images.cuda()
 
@@ -331,9 +331,9 @@ class DiNOv2(torch.nn.Module):
 
 dinov2 = DiNOv2()
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=60)
 def image_dino_feature(images, node_type="block", layer=-1):
-
+    global USE_CUDA
     if USE_CUDA:
         images = images.cuda()
 
@@ -430,10 +430,11 @@ class CLIP(torch.nn.Module):
 
 clip = CLIP()
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=60)
 def image_clip_feature(
     images, node_type="block", layer=-1
 ):
+    global USE_CUDA
     if USE_CUDA:
         images = images.cuda()