Update deepfake_vs_real.py

deepfake_vs_real.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/Deepfake-vs-Real-8000"
 model = SiglipForImageClassification.from_pretrained(model_name)
 processor = AutoImageProcessor.from_pretrained(model_name)
 
+@spaces.GPU
 def deepfake_classification(image):
     """Predicts whether an image is a Deepfake or Real."""
     image = Image.fromarray(image).convert("RGB")