AYYasaswini commited on
Commit
974b4f9
·
verified ·
1 Parent(s): c45f0f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ torch.manual_seed(1)
53
  logging.set_verbosity_error()
54
 
55
  # Set device
56
- torch_device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
57
  #if "mps" == torch_device: os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = "1"
58
 
59
  """## Loading the models
 
53
  logging.set_verbosity_error()
54
 
55
  # Set device
56
+ torch_device = "cpu"
57
  #if "mps" == torch_device: os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = "1"
58
 
59
  """## Loading the models