Update app.py

app.py

@@ -13,19 +13,16 @@ class Model:
         self.pipe_t2i = None
         self.pipe_i2i = None
 
-"""
-AssertionError: parameter: src must be one of dict_keys(['huggingface', 'models', 'spaces'])
-"""
 models = [
    Model("Marvel","models/ItsJayQz/Marvel_WhatIf_Diffusion", "whatif style"), 
    Model("Cyberpunk Anime Diffusion", "models/DGSpitzer/Cyberpunk-Anime-Diffusion", "dgs illustration style"),
-   Model("Guan Yu Diffusion", "models/DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu"),
+   Model("Guan Yu Diffusion", "models/DGSpitzer/Guan-Yu-Diffusion", "Guan-Yu style"),
    Model("Portrait plus", "models/wavymulder/portraitplus", "portrait+ style"),
    Model("classic Disney", "models/nitrosocke/classic-anim-diffusion", "classic disney style"),
-   Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
+   Model("SD21","models/stabilityai/stable-diffusion-2-1", "sd21 style")
 ]
 
-custom_model = models[3].path
+custom_model = "no style selected"
 
 def selectModel(message):
     message = message.lower()