Update app.py

app.py

@@ -31,9 +31,10 @@ models = [
 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)!=-1:
+        if message.find(models[i].prefix.lower())!=-1:
             c_model=models[i].path
             logging.warning('model selected = '+c_model)
             return c_model
@@ -51,7 +52,6 @@ sandbox = gr.Interface.load(
     api_key=API_KEY
 )
 
-logging.warning('prompt = '+inputs)
 logging.warning('model chosen = '+custom_model)
 
 sandbox.queue(concurrency_count=20).launch()