Upload app.py

app.py

@@ -11,26 +11,21 @@ elif("T4" in gpu_info):
 
 # os.chdir("/home/user/app/stable-diffusion-webui")
 
-os.system(f"pip install google-colab")
-# Mount our drive
-from google.colab import drive
-drive.mount('/content/drive')
-
 # Create a folder for storing files
 os.system(f"mkdir sd-webui-files")
 os.chdir("sd-webui-files")
 
 # # Clone SD WebUI from AUTOMATIC1111's repo
-# os.system(f"git clone --depth=1 https://github.com/AUTOMATIC1111/stable-diffusion-webui.git sd-webui-files/stable-diffusion-webui")
+os.system(f"git clone --depth=1 https://github.com/AUTOMATIC1111/stable-diffusion-webui.git sd-webui-files/stable-diffusion-webui")
 
 # # Download checkpoint models to /stable-diffusion-webui/models/Stable-diffusion
-# !wget -nc -P /content/drive/MyDrive/sd-webui-files/stable-diffusion-webui/models/Stable-diffusion https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.5-pruned.safetensors
+os.system(f"wget -nc -P stable-diffusion-webui/models/Stable-diffusion https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.5-pruned.safetensors")
+
 
-# # Fix change branch issues
-# %cd /content/drive/MyDrive/sd-webui-files/stable-diffusion-webui/
-# !git reset --hard
-# !git pull
-# !sed -i -e 's/checkout {commithash}/checkout --force {commithash}/g' launch.py
+os.system(f"git reset --hard")
+os.system(f"git pull")
+os.system(f"sed -i -e 's/checkout {commithash}/checkout --force {commithash}/g' launch.py")
 
+os.system(f"python launch.py --share --xformers --enable-insecure-extension-access --theme light --skip-torch-cuda-test")
 # Launch WebUI