Update app.py

app.py

@@ -27,21 +27,6 @@ args = parser.parse_args()
 
 # Use the original GPU ID throughout the entire code for consistency
 GPU_ID = args.gpu
-
-# Set environment variables - this used to remap the GPU, but we're removing this for consistency
-# Instead, we'll pass the original GPU ID to all commands
-# os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu)  # Commented out to ensure consistent GPU ID usage
-
-# Check if CUDA is available
-CUDA_AVAILABLE = torch.cuda.is_available()
-if CUDA_AVAILABLE:
-    GPU_COUNT = torch.cuda.device_count()
-    GPU_NAMES = [f"{i}: {torch.cuda.get_device_name(i)}" for i in range(GPU_COUNT)]
-else:
-    GPU_COUNT = 0
-    GPU_NAMES = ["CPU (CUDA not available)"]
-    GPU_ID = "CPU"
-
 DEFAULT_MODEL_PATH = args.model_path
 OUTPUT_DIR = args.output_dir