Update app.py

app.py

@@ -921,13 +921,7 @@ if __name__ == "__main__":
     # --- Model Loading ---
     # --- Model Loading ---set
     # Set device
-    if "cuda" in args.device and torch.cuda.is_available():
-        device = args.device
-        local_rank = 0  # Assume single GPU for Gradio unless configured otherwise
-        torch.cuda.set_device(local_rank)  # Set default CUDA device
-    else:
-        device = "cpu"
-        local_rank = -1  # Indicate CPU
+    device = args.device
     logging.info(f"Using device: {device}")
 
     args.torch_dtype = dict(fp16=torch.float16, fp32=torch.float32, bf16=torch.bfloat16)[args.torch_dtype]