Update app.py

app.py

@@ -17,24 +17,23 @@ models = [
    Model("SD21","models/stabilityai/stable-diffusion-2-1", "sd21 default style")
 ]
 
+model1=[]
 model2=[]
 model3=[]
 
 for i in range(len(models)):
     model3.append(models[i].name)
     model2.append(models[i].prefix)
+    model1.append(gradio.Interface.load(models[i].path))
 
 def process1(prompt):
-    modelSelected=''
+    modelSelected='models/stabilityai/stable-diffusion-2-1'
     for i in range(len(models)):
         if prompt.find(models[i].prefix)!=-1:
             modelSelected=models[i].path
-            print(modelSelected)
-            m=gradio.Interface.load(modelSelected)
-    if (modelSelected==''):
-        modelSelected = "models/stabilityai/stable-diffusion-2-1"
-        m=gradio.Interface.load(modelSelected)
-    image_return = m(prompt)
+    print(modelSelected)
+    model_idx=model3.index(modelSelected)
+    image_return = model1[model_idx](prompt)
     return image_return
 
 sandbox = gradio.Interface(fn=process1,