Update app.py

app.py

@@ -25,9 +25,20 @@ models = [
    Model("SD21","models/stabilityai/stable-diffusion-2-1", "")
 ]
 
-custom_model = models[3].path
+"""custom_model = models[3].path"""
+
+def selectModel(message):
+    message = message.lower()
+    for i in range(len(models)):
+        if message.find(models[i].prefix)!=-1:
+            c_model=models[i].path
+            return custom_model
+        else:
+            c_model=models[i].path
+    return c_model
 
 sandbox = gr.Interface.load(
+    selectModel,
     name= custom_model,
     title="""AlStable sandbox""",
     prompt = gr.Textbox(label="Prompt", show_label=False, max_lines=2, placeholder="Enter your prompt~", elem_id="input-prompt"), 
@@ -35,4 +46,7 @@ sandbox = gr.Interface.load(
     article=article,
     api_key=API_KEY
 )
+
+custom_model=sandbox.selectModel(prompt)
+
 sandbox.queue(concurrency_count=20).launch()