Update sketch_126.py

sketch_126.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from transformers import AutoImageProcessor, SiglipForImageClassification
 from transformers.image_utils import load_image
 from PIL import Image
@@ -9,6 +10,7 @@ model_name = "prithivMLmods/Sketch-126-DomainNet"
 model = SiglipForImageClassification.from_pretrained(model_name)
 processor = AutoImageProcessor.from_pretrained(model_name)
 
+@spaces.GPU
 def sketch_classification(image):
     """Predicts the sketch category for an input image."""
     # Convert the input numpy array to a PIL Image and ensure it has 3 channels (RGB)