Update app.py

adding more models

app.py

@@ -6,10 +6,26 @@ API_KEY=os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
 article = """---
 This space was created using [SD Space Creator](https://huggingface.co/spaces/anzorq/sd-space-creator)."""
 
+class Model:
+    def __init__(self, name, path="", prefix=""):
+        self.name = name
+        self.path = path
+        self.prefix = prefix
+        self.pipe_t2i = None
+        self.pipe_i2i = None
+
+models = [
+   Model("Marvel","ItsJayQz/Marvel_WhatIf_Diffusion", "Marvel")
+   Model("Cyberpunk Anime Diffusion", "DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style "),
+   Model("Guan Yu Diffusion", "DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu")
+]
+
+custom_model = "models/stabilityai/stable-diffusion-2-1"
+
 gr.Interface.load(
-    name="models/stabilityai/stable-diffusion-2-1",
+    name= custom_model,
     title="""Stable Diffusion 2 1 sandbox""",
     description="""Demo for <a href="https://huggingface.co/stabilityai/stable-diffusion-2-1">Stable Diffusion 2 1</a> Stable Diffusion model.""",
     article=article,
     api_key=API_KEY,
-    ).queue(concurrency_count=20).launch()
+    ).queue(concurrency_count=20).launch()