Create app.py

app.py

@@ -8,8 +8,7 @@ from transformers import (
 )
 
 import gradio as gr
-
-# import spaces  # ZERO GPU
+import spaces  # ZERO GPU
 
 
 MODEL_NAMES = ["p1atdev/wd-swinv2-tagger-v3-hf"]
@@ -90,7 +89,7 @@ def animagine_prompt(rating: list[str], character: list[str], general: list[str]
     return ", ".join(all_tags)
 
 
-# @spaces.GPU
+@spaces.GPU
 @torch.no_grad()
 def predict_tags(
     image: Image.Image, general_threshold: float = 0.3, character_threshold: float = 0.8