strongpear commited on
Commit
7bbedb1
1 Parent(s): f772b19

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -258,7 +258,7 @@ class inferSSCL():
258
  parser = argparse.ArgumentParser()
259
  parser.add_argument('--task', default='infer')
260
  parser.add_argument('--smooth_factor', type=float, default=0.9)
261
- device = 'cuda:0'
262
  emb_size = 256
263
 
264
  args = parser.parse_args(args=[])
 
258
  parser = argparse.ArgumentParser()
259
  parser.add_argument('--task', default='infer')
260
  parser.add_argument('--smooth_factor', type=float, default=0.9)
261
+ device = 'cpu'
262
  emb_size = 256
263
 
264
  args = parser.parse_args(args=[])