Update app.py

app.py

@@ -21,7 +21,7 @@ def run_dif(out_prompt,model_drop,cnt):
     out_box=[]
     out_html=""
     #for i,ea in enumerate(loaded_model):
-    for i in range(cnt):
+    for i in range(int(cnt)):
         p_seed+=" "
         try:
             model=loaded_model[int(model_drop)]
@@ -133,7 +133,7 @@ with gr.Blocks(css=css) as app:
         
     with gr.Row():
         model_drop=gr.Dropdown(label="Models", choices=models, type='index', value=models[0])
-        cnt = gr.Number()
+        cnt = gr.Number(value=1)
     out_html=gr.HTML()
     outp=gr.Gallery()
     btn.click(run_dif,[inp,model_drop,cnt],[outp,out_html])