added spaces.GPU at the classify_image

app.py

@@ -8,6 +8,8 @@ import gdown
 import numpy as np
 import tensorflow as tf
 
+import spaces
+
 from PIL import Image
 from tensorflow.keras.applications.mobilenet_v2 import preprocess_input
 
@@ -138,7 +140,7 @@ def preprocess_image(img):
 # ============================================================
 # PREDICTION
 # ============================================================
-
+@spaces.GPU
 def classify_image(img):
     """Return class probabilities and a human-readable result."""