Update app.py

app.py

@@ -25,7 +25,7 @@ pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion
 def load_model(model_drop):
     pipeline = AutoPipelineForImage2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float32, use_safetensors=True)
 
-def run_dif(prompt,im_path,model_drop,cnt):
+def run_dif(prompt,im_path,model_drop,cnt,strength,guidance,infer):
     print(f'im_path:: {im_path}')
     print(f'im_path0:: {im_path.root[0]}')
     print(f'im_path0.image.path:: {im_path.root[0].image.path}')
@@ -43,7 +43,7 @@ def run_dif(prompt,im_path,model_drop,cnt):
         
         # 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]
+        image = pipeline(prompt, image=init_image, strength=float(strength),guidance_scale=float(guidance),num_inference_steps=int(infer)).images[0]
         #make_image_grid([init_image, image], rows=1, cols=2)
         out_box.append(image)
         yield out_box,""
@@ -181,9 +181,13 @@ with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column():
             inp=gr.Textbox(label="Prompt")
+            strength=gr.Slider(label="Strength",minimum=0,maximum=1,step=0.1,value=0.2)
+            guidance=gr.Slider(label="Guidance",minimum=0,maximum=10,step=0.1,value=8.0)
+            infer=gr.Slider(label="Inference Steps",minimum=0,maximum=50,step=1,value=10)
+            
             with gr.Row():
                 btn=gr.Button()
-                stop_btn=gr.Button()
+                stop_btn=gr.Button("Stop")
         with gr.Column():
             inp_im=gr.Image(type='filepath')
             im_btn=gr.Button("Image Grid")
@@ -195,6 +199,6 @@ with gr.Blocks(css=css) as app:
     fingal=gr.Gallery(columns=10)
     im_list=gr.Textbox()    
     im_btn.click(load_im,inp_im,[outp,im_list])
-    go_btn = btn.click(run_dif,[inp,outp,model_drop,cnt],[fingal,out_html])
+    go_btn = btn.click(run_dif,[inp,outp,model_drop,cnt,strength,guidance,infer],[fingal,out_html])
     stop_btn.click(None,None,None,cancels=[go_btn])
 app.queue().launch()