Update app.py

app.py

@@ -1,8 +1,6 @@
 import gradio as gr
 import numpy as np
 import random
-import uuid
-from PIL import Image
 
 import spaces
 from diffusers import DiffusionPipeline
@@ -49,16 +47,7 @@ style_list = [
 STYLE_NAMES = [style["name"] for style in style_list]
 DEFAULT_STYLE_NAME = STYLE_NAMES[0]
 
-grid_sizes = {
-    "2x1": (2, 1),
-    "1x2": (1, 2),
-    "2x2": (2, 2),
-    "2x3": (2, 3),
-    "3x2": (3, 2),
-    "1x1": (1, 1)
-}
-
-@spaces.GPU(duration=100)
+@spaces.GPU
 def infer(
     prompt,
     negative_prompt="",
@@ -69,10 +58,9 @@ def infer(
     guidance_scale=0.0,
     num_inference_steps=4,
     style="Style Zero",
-    grid_size="1x1",
     progress=gr.Progress(track_tqdm=True),
 ):
-    
+    # Apply selected style
     selected_style = next(s for s in style_list if s["name"] == style)
     styled_prompt = selected_style["prompt"].format(prompt=prompt)
     styled_negative_prompt = selected_style["negative_prompt"]
@@ -82,33 +70,17 @@ def infer(
 
     generator = torch.Generator().manual_seed(seed)
 
-    grid_size_x, grid_size_y = grid_sizes.get(grid_size, (2, 2))
-    num_images = grid_size_x * grid_size_y
-
-    images = []
-    for _ in range(num_images):
-        image = pipe(
-            prompt=styled_prompt,
-            negative_prompt=styled_negative_prompt,
-            guidance_scale=guidance_scale,
-            num_inference_steps=num_inference_steps,
-            width=width,
-            height=height,
-            generator=generator,
-        ).images[0]
-        images.append(image)
-
-    # Create a grid image
-    grid_img = Image.new('RGB', (width * grid_size_x, height * grid_size_y))
-
-    for i, img in enumerate(images[:num_images]):
-        grid_img.paste(img, (i % grid_size_x * width, i // grid_size_x * height))
-
-    # Save the grid image
-    unique_name = str(uuid.uuid4()) + ".png"
-    grid_img.save(unique_name)
-    
-    return unique_name, seed
+    image = pipe(
+        prompt=styled_prompt,
+        negative_prompt=styled_negative_prompt,
+        guidance_scale=guidance_scale,
+        num_inference_steps=num_inference_steps,
+        width=width,
+        height=height,
+        generator=generator,
+    ).images[0]
+
+    return image, seed
 
 examples = [
     "Chocolate dripping from a donut against a yellow background, in the style of brocore, hyper-realistic oil --ar 2:3 --q 2 --s 750 --v 5  --ar 2:3 --q 2 --s 750 --v 5",
@@ -137,11 +109,11 @@ with gr.Blocks(css=css, theme="prithivMLmods/Minecraft-Theme") as demo:
                 placeholder="Enter your prompt",
                 container=False,
             )
-            
+
             run_button = gr.Button("Run", scale=0, variant="primary")
-                    
-        result = gr.Image(label="Result", show_label=False)
 
+        result = gr.Image(label="Result", show_label=False)
+        
         with gr.Row(visible=True):
             style_selection = gr.Radio(
                 show_label=True,
@@ -151,15 +123,8 @@ with gr.Blocks(css=css, theme="prithivMLmods/Minecraft-Theme") as demo:
                 value=DEFAULT_STYLE_NAME,
                 label="Quality Style",
             )
-            
-        with gr.Row(visible=False):
-            grid_size_selection = gr.Dropdown(
-                choices=["2x1", "1x2", "2x2", "2x3", "3x2", "1x1"],
-                value="1x1",
-                label="Grid Size"
-            )
 
-        with gr.Accordion("Advanced Settings", open=False, visibile=False):
+        with gr.Accordion("Advanced Settings", open=False, visible=False):
             negative_prompt = gr.Text(
                 label="Negative prompt",
                 max_lines=1,
@@ -211,11 +176,7 @@ with gr.Blocks(css=css, theme="prithivMLmods/Minecraft-Theme") as demo:
                     value=4,
                 )
 
-        gr.Examples(examples=examples, 
-                    inputs=[prompt], 
-                    outputs=[result, seed], 
-                    fn=infer, 
-                    cache_examples=True)
+        gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=True)
         
     gr.on(
         triggers=[run_button.click, prompt.submit],
@@ -230,10 +191,10 @@ with gr.Blocks(css=css, theme="prithivMLmods/Minecraft-Theme") as demo:
             guidance_scale,
             num_inference_steps,
             style_selection,
-            grid_size_selection,
         ],
         outputs=[result, seed],
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+