Update fashion_mnist_cloth.py

fashion_mnist_cloth.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from transformers import AutoImageProcessor
 from transformers import SiglipForImageClassification
 from transformers.image_utils import load_image
@@ -10,6 +11,7 @@ model_name = "prithivMLmods/Fashion-Mnist-SigLIP2"
 model = SiglipForImageClassification.from_pretrained(model_name)
 processor = AutoImageProcessor.from_pretrained(model_name)
 
+@spaces.GPU
 def fashion_mnist_classification(image):
     """Predicts fashion category for an image."""
     image = Image.fromarray(image).convert("RGB")