fix: model device

app.py

@@ -16,8 +16,8 @@ MODEL_NAME = MODEL_NAMES[0]
 
 model = AutoModelForImageClassification.from_pretrained(
     MODEL_NAME,
-    device="cuda",
 )
+model.to("cuda" if torch.cuda.is_available() else "cpu")
 processor = AutoImageProcessor.from_pretrained(MODEL_NAME, trust_remote_code=True)
 
 
@@ -90,7 +90,7 @@ def animagine_prompt(rating: list[str], character: list[str], general: list[str]
     return ", ".join(all_tags)
 
 
-@spaces.GPU
+@spaces.GPU(enable_queue=True)
 def predict_tags(
     image: Image.Image, general_threshold: float = 0.3, character_threshold: float = 0.8
 ):