feng2022 commited on
Commit
ca65a8e
1 Parent(s): f8b63c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ TOKEN = "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"
30
 
31
  def parse_args() -> argparse.Namespace:
32
  parser = argparse.ArgumentParser()
33
- parser.add_argument('--device', type=str, default='cuda:0')
34
  parser.add_argument('--theme', type=str)
35
  parser.add_argument('--live', action='store_true')
36
  parser.add_argument('--share', action='store_true')
 
30
 
31
  def parse_args() -> argparse.Namespace:
32
  parser = argparse.ArgumentParser()
33
+ parser.add_argument('--device', type=str, default='cpu')
34
  parser.add_argument('--theme', type=str)
35
  parser.add_argument('--live', action='store_true')
36
  parser.add_argument('--share', action='store_true')