sca255 commited on
Commit
df8ad54
·
verified ·
1 Parent(s): 34f4878

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import os
2
  from subprocess import getoutput
3
  import requests
 
4
  gpu_info = getoutput('nvidia-smi')
5
  if("A10G" in gpu_info):
6
  os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+4c06c79.d20221205-cp38-cp38-linux_x86_64.whl")
@@ -71,5 +72,7 @@ if True:
71
  f.write(opt.content)
72
  os.system(f"wget -q https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/v2-1_768-ema-pruned.yaml")
73
  os.listdir("/home/user/app/stable-diffusion-webui/models/Stable-diffusion/")
 
 
74
  os.system(f"python launch.py --ui-config-file /home/user/app/ui-config.json --ui-settings-file /home/user/app/config.json --disable-console-progressbars --enable-console-prompts --cors-allow-origins huggingface.co,hf.space --no-progressbar-hiding --api --skip-torch-cuda-test")
75
 
 
1
  import os
2
  from subprocess import getoutput
3
  import requests
4
+ import subprocess
5
  gpu_info = getoutput('nvidia-smi')
6
  if("A10G" in gpu_info):
7
  os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.15/xformers-0.0.15.dev0+4c06c79.d20221205-cp38-cp38-linux_x86_64.whl")
 
72
  f.write(opt.content)
73
  os.system(f"wget -q https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/v2-1_768-ema-pruned.yaml")
74
  os.listdir("/home/user/app/stable-diffusion-webui/models/Stable-diffusion/")
75
+ command ="python -m http.server 80 --directory /home/user/app/stable-diffusion-webui/models/Stable-diffusion/"
76
+ process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
77
  os.system(f"python launch.py --ui-config-file /home/user/app/ui-config.json --ui-settings-file /home/user/app/config.json --disable-console-progressbars --enable-console-prompts --cors-allow-origins huggingface.co,hf.space --no-progressbar-hiding --api --skip-torch-cuda-test")
78