Update app.py

app.py

@@ -18,23 +18,22 @@ class Model:
         self.prefix = prefix
         self.pipe_t2i = None
         self.pipe_i2i = None
-
 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 style"),
    Model("Portrait plus", "models/wavymulder/portraitplus", "portrait+ style"),
    Model("classic Disney", "models/nitrosocke/classic-anim-diffusion", "classic disney style"),
+   Model("vintedois", "models/22h/vintedois-diffusion-v0-1", "vintedois style"),
    Model("SD21","models/stabilityai/stable-diffusion-2-1", "sd21 default style")
 ]
 
 custom_model = "models/stabilityai/stable-diffusion-2-1"
 
 def selectModel(message):
-    logging.warning('message sent = '+message)
     message = message.lower()
     for i in range(len(models)):
-        if message.find(models[i].prefix.lower())!=-1:
+        if message.find(models[i].prefix)!=-1:
             c_model=models[i].path
             logging.warning('model selected = '+c_model)
             return c_model
@@ -52,6 +51,7 @@ sandbox = gr.Interface.load(
     api_key=API_KEY
 )
 
+logging.warning('prompt => '+inputs)
 logging.warning('model chosen = '+custom_model)
 
 sandbox.queue(concurrency_count=20).launch()