Update app.py

app.py

@@ -13,19 +13,16 @@ from tqdm.auto import tqdm
 import random
 import gradio as gr
 
-def generate_images(prompt, guidance_scale, n_samples, num_inference_steps):
-    seeds = [random.randint(1, 10000) for _ in range(n_samples)]
-    images = []
-    for seed in tqdm(seeds):
-        torch.manual_seed(seed)
-        image = pipe(prompt, num_inference_steps=num_inference_steps,guidance_scale=guidance_scale).images[0]
-        images.append(image)        
-    return images[0]
+def generate_image(prompt, guidance_scale, num_inference_steps):
+    seed = random.randint(1, 10000)
+    torch.manual_seed(seed)
+    image = pipe(prompt, num_inference_steps=num_inference_steps,guidance_scale=guidance_scale).images[0]
+    return image
 
-def gr_generate_images(prompt: str, num_images = 1, num_inference = 20, guidance_scale = 8 ):
+def gr_generate_images(prompt: str, num_inference = 20, guidance_scale = 8 ):
     prompt = prompt + "sks style"
-    images = generate_images(prompt, guidance_scale, num_images, num_inference)
-    return images
+    image = generate_image(prompt, guidance_scale, num_inference)
+    return image
 
 with gr.Blocks() as demo:
     examples = [
@@ -90,10 +87,6 @@ with gr.Blocks() as demo:
             height="50%",  # Use the full view height
             object_fit="scale-down"  # Maintain aspect ratio without cropping
         )
-    num_images_slider = 1
-    num_inference_steps_slider = 20
-    guidance_slider = 8
-
     btn.click(gr_generate_images, [text], gallery)
     gr.Examples(examples, inputs=[text])
     gr.HTML(