Update app.py

app.py

@@ -27,17 +27,20 @@ def load_model(model_drop):
 
 def run_dif(prompt,im_path,model_drop,cnt):
     out_box=[]
-    url = base_url+im_path
-    init_image = load_image(url)
     
-    #prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
-    
-    # pass prompt and image to pipeline
-    #image = pipeline(prompt, image=init_image, strength=0.8,guidance_scale=8.0,negative_prompt=negative_prompt,num_inference_steps=50).images[0]
-    image = pipeline(prompt, image=init_image, strength=0.8,guidance_scale=8.0,num_inference_steps=2).images[0]
-    #make_image_grid([init_image, image], rows=1, cols=2)
-    out_box.append(image)
-    return out_box,""
+    for ea in im_path:
+        print(ea)        
+        url = base_url+ea
+        init_image = load_image(url)
+        
+        #prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+        
+        # pass prompt and image to pipeline
+        #image = pipeline(prompt, image=init_image, strength=0.8,guidance_scale=8.0,negative_prompt=negative_prompt,num_inference_steps=50).images[0]
+        image = pipeline(prompt, image=init_image, strength=0.2,guidance_scale=8.0,num_inference_steps=2).images[0]
+        #make_image_grid([init_image, image], rows=1, cols=2)
+        out_box.append(image)
+        yield out_box,""
 
 
 
@@ -171,16 +174,20 @@ with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column():
             inp=gr.Textbox(label="Prompt")
-            btn=gr.Button()
+            with gr.Row():
+                btn=gr.Button()
+                stop_btn=gr.Button()
         with gr.Column():
             inp_im=gr.Image(type='filepath')
-            im_btn=gr.Button()
+            im_btn=gr.Button("Image Grid")
     with gr.Row():
         model_drop=gr.Dropdown(label="Models", choices=models, type='index', value=models[0])
         cnt = gr.Number(value=1)
     out_html=gr.HTML()
     outp=gr.Gallery(columns=10)
-
+    fingal=gr.Gallery(columns=10)
+    
     im_btn.click(load_im,inp_im,outp)
-    btn.click(run_dif,[inp,inp_im,model_drop,cnt],[outp,out_html])
+    go_btn = btn.click(run_dif,[inp,outp,model_drop,cnt],[fingal,out_html])
+    stop_btn.click(None,None,None,cancels='go_btn')
 app.launch()