Update app.py
Browse files
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()
|