inference on default dtype and available device

app.py

@@ -30,8 +30,8 @@ from transformers import AutoProcessor
 repo_id = "microsoft/OmniParser-v2.0"  # HF repo
 local_dir = "weights"  # Target local directory
 
-dtype = torch.bfloat16
-DEVICE = torch.device('cuda')  
+dtype = torch.get_default_dtype()
+DEVICE = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
 som_generator = MarkHelper()
 magma_som_prompt = "<image>\nIn this view I need to click a button to \"{}\"? Provide the coordinates and the mark index of the containing bounding box if applicable."