add more model options

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 from pipeline import CustomPipeline, setup_scheduler
+from diffusers import StableDiffusionPipeline
 from PIL import Image
 # from easydict import EasyDict as edict
 
@@ -106,8 +107,14 @@ if __name__ == "__main__":
     buttons = {}
 
     list_models = [
-
+        "Linaqruf/anything-v3.0",
+        "runwayml/stable-diffusion-v1-5",
+        "dreamlike-art/dreamlike-photoreal-2.0",
     ]
+    for model_id in list_models:
+        pipeline = StableDiffusionPipeline.from_pretrained(model_id)
+        del pipeline
+        print(f"Downloaded {model_id}")
 
     with gr.Blocks() as demo:
         gr.Markdown(