Hugging Face's logo

Spaces:
DReAMy-lib
/
dream
Sleeping

App Files Community
lorenzoscottb commited on
Commit
beb45b6
•
1 Parent(s): 1ba6f25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,7 +76,8 @@ def anon(text):
76
  anon_dreams_list = dreamy.anonimise(
77
  [text],
78
  return_original=False,
79
- batch_size=1
 
80
  )
81
 
82
  return anon_dreams_list[0]
 
76
  anon_dreams_list = dreamy.anonimise(
77
  [text],
78
  return_original=False,
79
+ batch_size=1,
80
+ device='cpu'
81
  )
82
 
83
  return anon_dreams_list[0]