Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import subprocess
2
 
3
  if __name__ == "__main__":
4
- cmd = "iopaint start --model lama --host 0.0.0.0 --port 7860"
5
  # start subprocess using cmd
6
  subprocess.run(cmd, shell=True)
 
1
  import subprocess
2
 
3
  if __name__ == "__main__":
4
+ cmd = "iopaint start --model lama --host 0.0.0.0 --port 7860 --device cuda"
5
  # start subprocess using cmd
6
  subprocess.run(cmd, shell=True)