Update app.py

app.py

@@ -26,35 +26,112 @@ scheduler = Scheduler.from_pretrained(model_path, subfolder="scheduler")
 pipe = Pipeline(vq_model, tokenizer=tokenizer, text_encoder=text_encoder, transformer=model, scheduler=scheduler)
 pipe.to(device)
 
-@spaces.GPU
-def generate_image(prompt, negative_prompt, resolution, steps, cfg, progress=gr.Progress(track_tqdm=True)):
+
+def generate_image(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+    if randomize_seed or seed == 0:
+        seed = torch.randint(0, MAX_SEED, (1,)).item()
+    torch.manual_seed(seed)
+    
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
-        height=resolution,
-        width=resolution,
-        guidance_scale=cfg,
-        num_inference_steps=steps
+        height=height,
+        width=width,
+        guidance_scale=guidance_scale,
+        num_inference_steps=num_inference_steps
     ).images[0]
     
-    return image
+    return image, seed
 
 # Default negative prompt
 default_negative_prompt = "worst quality, normal quality, low quality, low res, blurry, distortion, text, watermark, logo, banner, extra digits, cropped, jpeg artifacts, signature, username, error, sketch, duplicate, ugly, monochrome, horror, geometry, mutation, disgusting, bad anatomy, bad proportions, bad quality, deformed, disconnected limbs, out of frame, out of focus, dehydrated, disfigured, extra arms, extra limbs, extra hands, fused fingers, gross proportions, long neck, jpeg, malformed limbs, mutated, mutated hands, mutated limbs, missing arms, missing fingers, picture frame, poorly drawn hands, poorly drawn face, collage, pixel, pixelated, grainy, color aberration, amputee, autograph, bad illustration, beyond the borders, blank background, body out of frame, boring background, branding, cut off, dismembered, disproportioned, distorted, draft, duplicated features, extra fingers, extra legs, fault, flaw, grains, hazy, identifying mark, improper scale, incorrect physiology, incorrect ratio, indistinct, kitsch, low resolution"
 
-# Gradio interface
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[
-        gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
-        gr.Textbox(label="Negative Prompt", value=default_negative_prompt),
-        gr.Slider(512, 1024, 1024, step=64, label="Resolution"),
-        gr.Slider(1, 100, 48, step=1, label="Number of Steps"),
-        gr.Slider(1, 20, 9, step=0.5, label="CFG Scale")
-    ],
-    outputs=gr.Image(type="pil"),
-    title="Meissonic Image Generator",
-    description="Generate images using the Meissonic model. Enter a prompt and adjust parameters to create your image.",
-)
+css = """
+#col-container {
+    margin: 0 auto;
+    max-width: 640px;
+}
+"""
+
+examples = [
+    "a beautiful landscape painting with a river and mountains in the background",
+    "a futuristic cityscape at night with flying cars",
+    "a portrait of a smiling old man with wrinkles and wisdom in his eyes",
+]
+
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown("# Meissonic Text-to-Image Generator")
+        with gr.Row():
+            prompt = gr.Text(
+                label="Prompt",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter your prompt",
+                container=False,
+            )
+            run_button = gr.Button("Run", scale=0, variant="primary")
+        result = gr.Image(label="Result", show_label=False)
+        with gr.Accordion("Advanced Settings", open=False):
+            negative_prompt = gr.Text(
+                label="Negative prompt",
+                max_lines=1,
+                placeholder="Enter a negative prompt",
+                value=default_negative_prompt,
+            )
+            seed = gr.Slider(
+                label="Seed",
+                minimum=0,
+                maximum=MAX_SEED,
+                step=1,
+                value=0,
+            )
+            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+            with gr.Row():
+                width = gr.Slider(
+                    label="Width",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
+                height = gr.Slider(
+                    label="Height",
+                    minimum=256,
+                    maximum=MAX_IMAGE_SIZE,
+                    step=32,
+                    value=1024,
+                )
+            with gr.Row():
+                guidance_scale = gr.Slider(
+                    label="Guidance scale",
+                    minimum=0.0,
+                    maximum=20.0,
+                    step=0.1,
+                    value=9.0,
+                )
+                num_inference_steps = gr.Slider(
+                    label="Number of inference steps",
+                    minimum=1,
+                    maximum=100,
+                    step=1,
+                    value=48,
+                )
+        gr.Examples(examples=examples, inputs=[prompt])
+    gr.on(
+        triggers=[run_button.click, prompt.submit],
+        fn=generate_image,
+        inputs=[
+            prompt,
+            negative_prompt,
+            seed,
+            randomize_seed,
+            width,
+            height,
+            guidance_scale,
+            num_inference_steps,
+        ],
+        outputs=[result, seed],
+    )
 
-iface.launch()
+demo.launch()