Upload 2 files

README.md

@@ -1,12 +1,11 @@
 ---
 title: Mountdrive
-emoji: 🐠
-colorFrom: purple
-colorTo: purple
+emoji: 🌍
+colorFrom: blue
+colorTo: blue
 sdk: gradio
-sdk_version: 3.39.0
+sdk_version: 3.9
 app_file: app.py
-pinned: false
+pinned: true
+duplicated_from: 
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,35 @@
+import os
+from subprocess import getoutput
+
+gpu_info = getoutput('nvidia-smi')
+if("A10G" in gpu_info):
+    os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp38-cp38-linux_x86_64.whl")
+elif("T4" in gpu_info):
+    os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp38-cp38-linux_x86_64.whl")
+
+# os.system(f"git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git /home/user/app/stable-diffusion-webui")
+
+# os.chdir("/home/user/app/stable-diffusion-webui")
+
+# 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")
+
+# # 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
+
+# # 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
+
+# Launch WebUI
+