Update app.py

app.py

@@ -31,7 +31,8 @@ class NecklaceTryOn:
     def clear_func(self):
         torch.cuda.empty_cache()
         gc.collect()
-
+        
+    @spaces.GPU
     def apply_necklace(self, image, jewellery):
         """Apply necklace on the image and return modified image and mask."""
         image = np.array(image)