AlStable commited on
Commit
a6ca70e
1 Parent(s): a9216e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,10 +27,10 @@ for i in range(len(models)):
27
  model1.append(gradio.Interface.load(models[i].path))
28
 
29
  def process1(prompt):
30
- modelSelected='sd21 default style'
31
  for i in range(len(models)):
32
  if prompt.find(models[i].prefix)!=-1:
33
- modelSelected=models[i].path
34
  print(modelSelected)
35
  model_idx=model3.index(modelSelected)
36
  image_return = model1[model_idx](prompt)
 
27
  model1.append(gradio.Interface.load(models[i].path))
28
 
29
  def process1(prompt):
30
+ modelSelected=model3[0]
31
  for i in range(len(models)):
32
  if prompt.find(models[i].prefix)!=-1:
33
+ modelSelected=models[i].name
34
  print(modelSelected)
35
  model_idx=model3.index(modelSelected)
36
  image_return = model1[model_idx](prompt)