Omnibus commited on
Commit
6b28599
1 Parent(s): e122d23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -17,13 +17,15 @@ for i,model in enumerate(models):
17
  print (loaded_model)
18
 
19
  def run_dif(out_prompt,model_drop,cnt):
 
20
  out_box=[]
21
  out_html=""
22
  #for i,ea in enumerate(loaded_model):
23
  for i in range(cnt):
 
24
  try:
25
  model=loaded_model[int(model_drop)]
26
- out_img=model(out_prompt)
27
  print(out_img)
28
  out_box.append(out_img)
29
  except Exception as e:
@@ -130,7 +132,7 @@ with gr.Blocks(css=css) as app:
130
  btn=gr.Button()
131
 
132
  with gr.Row():
133
- model_drop=gr.Dropdown(label="Models", choices=models, type=index, value=models[0])
134
  cnt = gr.Number()
135
  out_html=gr.HTML()
136
  outp=gr.Gallery()
 
17
  print (loaded_model)
18
 
19
  def run_dif(out_prompt,model_drop,cnt):
20
+ p_seed=""
21
  out_box=[]
22
  out_html=""
23
  #for i,ea in enumerate(loaded_model):
24
  for i in range(cnt):
25
+ p_seed+=" "
26
  try:
27
  model=loaded_model[int(model_drop)]
28
+ out_img=model(out_prompt+p_seed)
29
  print(out_img)
30
  out_box.append(out_img)
31
  except Exception as e:
 
132
  btn=gr.Button()
133
 
134
  with gr.Row():
135
+ model_drop=gr.Dropdown(label="Models", choices=models, type='index', value=models[0])
136
  cnt = gr.Number()
137
  out_html=gr.HTML()
138
  outp=gr.Gallery()