cuda check

app.py

@@ -67,7 +67,7 @@ async def load_model():
     login(token=hf_token)
     
     # Load model (will cache in /home/user/.cache/huggingface)
-    
+    DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
     model = QwenClassifier.from_pretrained(
         hf_repo,
     ).to(DEVICE)