Fix device detection

Signed-off-by: Aivin V. Solatorio <[email protected]>

app.py

@@ -28,7 +28,7 @@ def get_model(model_name: str = None):
     if _MODEL.get(model_name) is None:
         _MODEL[model_name] = GLiNER.from_pretrained(model_name, cache_dir=_CACHE_DIR)
 
-    if torch.cuda.is_available() and _MODEL[model_name].device.type.startswith("cuda"):
+    if torch.cuda.is_available() and not next(_MODEL[model_name].parameters()).device.type.startswith("cuda"):
         _MODEL[model_name] = _MODEL[model_name].to("cuda")
 
     print(f"{datetime.now()} :: get_model :: {datetime.now() - start}")