Update app.py

app.py

@@ -79,7 +79,7 @@ def to_Canny(image):
 
     return canny_image
 
-def inference(prompt,canny_image,number,seed ):
+def inference(prompt,canny_image,number,seed, steps ):
     print("start inference")
 
     
@@ -94,29 +94,16 @@ def inference(prompt,canny_image,number,seed ):
     image_ = canny_image
     prompt = prompt
     out_image = pipe(
-      prompt, num_inference_steps=20, generator=generator, image=image_, num_images_per_prompt=number)
+      prompt, num_inference_steps=steps, generator=generator, image=image_, num_images_per_prompt=number)
     print('end inference')
     return out_image
     
-def generate_images(image, prompt):
-    pose = get_pose(image)
-    output = pipe(
-        prompt,
-        pose,
-        generator=generator,
-        num_images_per_prompt=3,
-        num_inference_steps=20,
-    )
-    all_outputs = []
-    all_outputs.append(pose)
-    for image in output.images:
-        all_outputs.append(image)
-    return all_outputs
 
-def generation(prompt,text,seed,police_size, lenght, width,number):
+
+def generation(prompt,text,seed,police_size, lenght, width,number,num_inference_steps):
     img = generate_an_image_from_text(text,police_size,lenght,width)
     img = to_Canny(img)
-    output = inference(prompt,img, number,seed)
+    output = inference(prompt,img, number,seed,num_inference_steps)
     all_outputs = []
     for image in output.images:
         all_outputs.append(image)
@@ -124,7 +111,7 @@ def generation(prompt,text,seed,police_size, lenght, width,number):
 
 gr.Interface(fn=generation, 
              inputs=["text", "text", gr.Slider(0, 200), gr.Slider(0, 200), gr.Slider(0, 1024), gr.Slider(0, 1024),
-                     gr.Slider(0, 7)], outputs=gr.Gallery().style(grid=[2], height="auto"), title="Generate a logo using Text ", examples=[["A steampunk Alphabetic Logo, steampunk style, with glowing mecha parts, mecha alphabets, high quality, high res, ultra HD", "Logo",60,90,512,512,2]]).launch(enable_queue=True)
+                     gr.Slider(0, 7),gr.Slider(0, 20)], outputs=gr.Gallery().style(grid=[2], height="auto"), title="Generate a logo using Text ", examples=[["A steampunk Alphabetic Logo, steampunk style, with glowing mecha parts, mecha alphabets, high quality, high res, ultra HD", "Logo",60,90,512,512,2]]).launch(enable_queue=True)