Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 from torchvision import transforms
 from transformers import AutoModelForImageClassification, AutoConfig
 import gradio as gr
+import spaces
 
 model_id = "thelabel/240903-image-tagging"
 config = AutoConfig.from_pretrained(model_id)
@@ -26,6 +27,7 @@ def load_image_from_url(url):
     except Exception as e:
         return None
 
+@spaces.GPU
 def predict_tags(image_url, threshold=0.5):
     image = load_image_from_url(image_url)
     if image is None: