close #112

nadermx · Dec 6, 2023 · 4f389f4 · 4f389f4
1 parent 1294514
commit 4f389f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backgroundremover/bg.py b/backgroundremover/bg.py
@@ -25,7 +25,7 @@
  else:
  DEVICE = torch.device('cpu')
 except Exception as e:
- print(f"An error occurred: {e}")
+ print(f"Using CPU. Setting Cuda or MPS failed: {e}")
  DEVICE = torch.device('cpu')
 
 class Net(torch.nn.Module):