Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 negative = "(worst quality, low quality:1.4), (bad-image-v2-39000:0.75), (bad_prompt_v2:0.85)"
 sampler = "DPM++ 2M Karras"
 steps = 35
-scale = 7.0
+scale = 8.0
 
 embeddings = [
     "https://huggingface.co/datasets/Nerfgun3/bad_prompt/resolve/main/bad_prompt_version2.pt",
@@ -19,17 +19,6 @@ models = [
     "https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0.vae.pt",
 ]
         
-default_model = None
-default_vae = None
-                
-for model in models:
-    if None not in (default_model, default_vae) or (default_model and 'vae.pt' not in ''.join(models)):
-        break
-    if "vae.pt" in model:
-        default_vae = model
-    else:
-        default_model = model
-
 os.system("git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui")
 os.chdir("stable-diffusion-webui")
     
@@ -40,6 +29,9 @@ os.system(f"wget {' '.join(embeddings)}")
 
 os.chdir("../models/Stable-diffusion")
 os.system(f"wget {' '.join(models)}")
+files = os.listdir()
+default_vae = [f for f in files if f.endswith(".vae.pt")][0]
+default_model = [f for f in files if f.endswith(".safetensors") or f.endswith(".ckpt")][0]
 os.chdir("../..")
 
 os.system("pip install -U -qq pip")