Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 max_img_wh = 4096  # Set the maximum image size in pixels
 
 def inference(image, use_gpu):
-    gpu_ids = 'cuda' if use_gpu else ''
+    gpu_ids = '0' if use_gpu else ''
     model = ModelLoader(gpu_ids=gpu_ids, max_img_wh=max_img_wh)
     model.load()
     output_img = model.inference(image_pil=image)