Omnibus commited on
Commit
9992f07
1 Parent(s): 028d3c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,7 +33,8 @@ def run_dif_color(out_prompt,model_drop,cnt,color,tint):
33
  for i in range(int(cnt)):
34
  p_seed+=" "
35
  try:
36
- model=loaded_model[int(model_drop)]
 
37
  out_img=model(out_prompt+p_seed)
38
  print(out_img)
39
 
 
33
  for i in range(int(cnt)):
34
  p_seed+=" "
35
  try:
36
+ model=gr.load(f'models/{model[int(model_drop)]}')
37
+ #model=loaded_model[int(model_drop)]
38
  out_img=model(out_prompt+p_seed)
39
  print(out_img)
40