Kremon96 commited on
Commit
62b2710
·
verified ·
1 Parent(s): f35045e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -181,7 +181,7 @@ def load_models():
181
  try:
182
  import silentcipher
183
  # print("Loading silentcipher watermarking model...")
184
- silentcipher_model = silentcipher.get_model(model_type='44.1k', device='cuda')
185
  # print("Silentcipher model loaded successfully!")
186
  except Exception as e:
187
  print(f"Warning: Failed to load silentcipher model: {e}")
 
181
  try:
182
  import silentcipher
183
  # print("Loading silentcipher watermarking model...")
184
+ silentcipher_model = silentcipher.get_model(model_type='44.1k', device='cpu')
185
  # print("Silentcipher model loaded successfully!")
186
  except Exception as e:
187
  print(f"Warning: Failed to load silentcipher model: {e}")